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

思路

暴力解的，就是按升序合并两个数组，存到新数组c中，然后再分情况判断c的长度是奇数还是偶数，奇数则中位数就是c[len/2]，偶数则中位数是(c[len/2]+c[len/2-1])/2.0000，注意因为输出是double类型，所以要除2.0000

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int a=nums1.size();
        int b=nums2.size();
        int len=a+b;
        int n=0,m=0,i=0;
        int c[len];                  //定义一个新数组去合并两个数组
        while(n<a&&m<b){             //两个数组都没有被遍历完
            if(nums1[n]<nums2[m]){   //较小的先放进数组c
                c[i++]=nums1[n++];
            }
            else c[i++]=nums2[m++];
        }
        while(m<b){
            //第二个数组还没有被遍历完，则直接将剩余的项按顺序放到新数组
            c[i++]=nums2[m++];
        }
        while(n<a){
            c[i++]=nums1[n++];
        }
        double mid=0.0000;      //注意输出的是double类型
        if(len%2){              //数组长度为奇数，则中位数为中间那项
            mid=c[len/2];
        }
        else mid=(c[len/2]+c[len/2-1])/2.0000;
        //数组长度为偶数，则中位数为中间两项的平均值
        return mid;
    }
};