【LeetCode Hot100 C++】（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

解题思路

本题需要求两个有序数列的中位数，我们直到中位数的求法，是根据整个数列的长度来确定的。因此，需进行两种求法，一种为总长度为奇数、一种为偶数。直接写有些繁琐，因此，我们需要先写一个辅助函数，求两个数列第k小的数。

---

求两个数列第k小的数函数的思路

直接法

一种比较好想到的方法，是直接法，设立两指针分布指向两数组头部，谁小移动谁，每次移动令k-1，若一个指针移动到了尾部，则让另一个指针加上剩余的k值。这种方法比较简单，但时间复杂度为O（M+N）。

二分法

基本思想是，比较两个数组第（k/2）个元素的大小，哪个数组比较的值小，则丢弃掉那个数组前k/2个数，使成为一个新的数组，k也需要减去丢弃的数。这样不断循环迭代，若一个数组把数已经丢完了，若k还有，则将k加到另一个数的指针上，或者是k为1是，则比较返回两个指针指向中较小的一个。下面我演示一下具体过程。

我将演示求以下两个数组中第5小的数，先设置两个指针指向头部。

 

 第一个指针加上k/2-1,指向了2，第二个加上指向了3

 

明显该舍弃蓝色指针前的数，此时k变成了3,记住每次比较完，指针都要重置回头部

 

 

 

接着继续比较，让蓝色指针加上k/2-1，发现指向了4，而红色指针指向了0

 

 故删除第二个数组元素，此时k变为2

 

 继续比较，蓝色指向4，红色指向3

删除后，K变为1 

此时由于k为1满足边界条件，返回两指针指向数组中较小的数。

---

代码如下 

class Solution {
public:

    double FindkthNum(vector<int>& nums1, vector<int>& nums2,int k){
        int len = nums1.size()+nums2.size();
        int i1 = 0,i2 = 0;//i1,i2始终为指向两数组头的指针
        int n1 = nums1.size();
        int n2= nums2.size();
     
        while(true){
                //3if为边界条件
                if(i1 == n1){
                    return nums2[i2 + k - 1];
                }
                if(i2 == n2){
                    return nums1[i1 + k - 1];
                }
                if(k == 1){
                    return min(nums1[i1],nums2[i2]);
                }
                //比较指针位置
                int newi1 = min(i1 + k/2 - 1,n1 - 1);
                int newi2 = min(i2 + k/2 - 1,n2 - 1);
                //比较大小
                if(nums1[newi1] <= nums2[newi2]){
                    k -= newi1 - i1 + 1;//变化k值，因为回删除newi1 - i1 + 1个值
                    i1 = newi1 + 1;//相当于删除了数组前面元素
                    
                }
                else{
                    k -= newi2 - i2 + 1;
                    i2 = newi2 + 1;
                }

            
        }
    }
    //求中位数
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len = nums1.size()+nums2.size();
        int k =len/2;
        int k2 = len/2 + 1;
        if(len %2 == 1){
            return FindkthNum(nums1,nums2,k2);
        }
        else{
            return (FindkthNum(nums1,nums2,k)+FindkthNum(nums1,nums2,k+1))/2;
        }
    }
};