Leetcode||4. Median of Two Sorted Arrays

要求时间复杂度为O(m+n),遍历一遍两个数组,将他们按顺序存入新开的数组。(类似归并)。然后输出。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1=nums1.size();
        int len2=nums2.size();
        int i=0,j=0;
        std::vector<int> array;
        while(i<len1&&j<len2)
        {
            if(nums1[i]<=nums2[j])
            {
                array.push_back(nums1[i]);
                i++;
            }else
            {
                array.push_back(nums2[j]);
                j++;
            }
        }
        while(i<len1)
        {
            array.push_back(nums1[i]);
            i++;
        }
        while(j<len2)
        {
            array.push_back(nums2[j]);
            j++;
        }
        if(array.size()%2==1)
        {
            return (double)array[array.size()/2];
        }
        else
        {
            return (double)(array[(array.size()-1)/2]+array[(array.size()+1)/2])/2;
        }
    }
};

另,看到zxzxy1988的二分方法很棒。

猜你喜欢

转载自blog.csdn.net/sinat_35205772/article/details/52664352