LeetCode - Median of Two Sorted Arrays / C++

문제 링크 : https://leetcode.com/problems/median-of-two-sorted-arrays/

Median of Two Sorted Arrays - LeetCode

 

 

풀이

정렬된 두 배열이 주어질 때 그 배열을 합친 뒤 다시 정렬하고 중간값을 리턴하면 된다.

이 때, size가 홀수라면 단순히 가운데 값을 리턴하면 되고

짝수라면 가운데 값과 가운데 이전 값의 평균을 리턴하면 된다.

 

 

코드

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> merged;
        
        for(int i=0; i<nums1.size(); i++){
            merged.push_back(nums1[i]);
        }
        for(int i=0; i<nums2.size(); i++){
            merged.push_back(nums2[i]);
        }
        
        sort(merged.begin(), merged.end());
        
        int mid = merged.size()/2;
            
        if(merged.size() % 2 == 1) return double(merged[mid]);
        else return (merged[mid] + double(merged[mid - 1])) / 2;
    }
};