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

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

这个其实刚开始我的思路是将这两个vector合成一个并排序，然后取出其中的中间索引的元素，并根据奇偶位返回相应的中位数。但是这两个数组都是有序的，用sort排序，快速排序在面对这样的情况时，效率并不理想。所以，我为了效率并没有使用这样的方法，而是遍历这两个数组，每次将计数器加一，直到计数器到达中间值。这样有比较好的空间复杂度

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        bool is_odd=(nums1.size()+nums2.size())%2==0?false:true;
        int mid1=0,mid2=0;
        int p=0,q=0;
        int tar=(nums1.size()+nums2.size())/2;
        while(p+q<=tar){
            mid1=mid2;
            if(p>=nums1.size()){
                mid2=nums2[q];
                q++;
                continue;
            }
             if(q>=nums2.size()){
                mid2=nums1[p];
                p++;
                continue;
            }
            if(nums1[p]>nums2[q]){
                mid2=nums2[q];
                q++;
            }
            else{
                mid2=nums1[p];
                p++;
            }
        }
        if(is_odd){return mid2;}
        else{return (double)(mid1+mid2)/2;}
    }
};

后来看到一个博客用STL的marge排序为两个有序数组合并并排序，于是就有了现在的这个方法，和我刚开始的想法是类似的，这个方法有比较优秀的时间复杂度

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        bool is_odd=(nums1.size()+nums2.size())%2==0?false:true;
        int mid1=0,mid2=0;
        int p=0,q=0;
        vector<int> nums3;
        nums3.resize(nums1.size()+nums2.size());
        merge(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),nums3.begin());
        mid2=nums3[nums3.size()/2];
        if(is_odd){return mid2;}
        else{
            mid1=nums3[nums3.size()/2-1];
            return (double)(mid1+mid2)/2;
            }
    }
};