LeetCode 4 寻找两个有序数组的中位数

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

以下给定两种解决方案：一种是合并两个有序数组，使它们成为一个有序数组；另一种则是不合并，利用两个指针分别指向两个有序数组的第一个元素，然后逐个比较大小。

c++代码

1. 合并两个数组的方法

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

2. 不合并数组的方法

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int mid1 = 0, mid2 = 0;
        vector<int>::iterator iter1 = nums1.begin(), iter2 = nums2.begin();
        for(int i = 0; i <= (nums1.size() + nums2.size())/2; i++) {
            mid1 = mid2;
            if(iter1 == nums1.end()){
                mid2 = *iter2;
                iter2++;
            }
            else if(iter2 == nums2.end()) {
                mid2 = *iter1;
                iter1++;
            }
            else if(*iter1 > *iter2) {
                mid2 = *iter2;
                iter2++;
            }
            else {
                mid2 = *iter1;
                iter1++;
            }
        }
        return (nums1.size() + nums2.size())%2? mid2:(mid1 + mid2)/2.0;
    }
};