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

【问题】给定两个大小为 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

【思路】比较简单的思路使用归并排序，为什么有这个思路呢？因为题目说两个数组是有序数组，因此我们对两个数组进行merge，如果小的数则放入res数组中，直到res的数组大小为(m*n)/2+1，因此最后在总个数为偶数时，中位数为res中最后两个数求平均，否则中位数为res数组中最后一个数。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int m = nums1.size(), n = nums2.size();
        int mid = (m + n) / 2 + 1;
        int i, j;
        for(i = 0, j = 0;i <m && j < n;){
            if(nums1[i] > nums2[j]){
                res.push_back(nums2[j++]);
            }
            else if(nums1[i] <= nums2[j]){
                res.push_back(nums1[i++]);
            }
            if(res.size() == mid){
                break;
            }
        }
        while(res.size() != mid){
            if(i < m)
                res.push_back(nums1[i++]);
            if(j < n)
                res.push_back(nums2[j++]);
        }

        int len = res.size();
        if((m + n) % 2 == 0)
            return (res[len-1] + res[len-2]) / 2.0;
        return res[len-1];
    }
};