leetcode 4 寻找两个正序数组的中位数

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

 

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2
示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5
示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000
示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000
示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000
 

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

一开始用vector存结果，但是用时过长，改为数组后快一倍。

先比较大小，将两个数组的内容合并为一个。再从这个数组里面寻找中位数，

合并的时候要注意当其中一个数组为空的情况，当其中一个为空，则直接复制另外一个数组就是结果数组。

如果都不为空：两个数组的启示索引都是0，合并一个则对应数组索引加1。、

每次合并完需要考虑：如果其中一个数组全部合并完了，就与其中一个为空的情况一样，直接复制另外一个数组剩下的元素。

 

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1=nums1.size();
        int len2=nums2.size();
        int res[2000]={0};
        int k=0;
        if(len1==0){
            for(int i=0;i<len2;i++){
                res[k++]=nums2[i];
            }
        }
        else if(len2==0){
            for(int i=0;i<len1;i++){
                res[k++]=nums1[i];
            }
        }
        else{
            int t1,t2;
            int lens=len1+len2;
            for(int i=0,j=0;k<lens;){
                if(nums1[i]<=nums2[j]){
                    res[k++]=nums1[i++];
                }else if(nums1[i]>nums2[j]){
                    res[k++]=nums2[j++];
                }
                if(i==len1){
                    for(int l=j;l<len2;l++){
                        res[k++]=nums2[l];
                    }
                    break;
                }else if(j==len2){
                    for(int l=i;l<len1;l++){
                        res[k++]=nums1[l];
                    }
                    break;
                }
            }
        }
        int lenres=k;
        if(lenres%2==0){
            double r=res[lenres/2-1]+res[(lenres/2)];
            return r/2.0;
        }else{
            return double(res[(lenres+1)/2-1]);
        }
        return 0.0;
    }
};