【Leetcode】 寻找两个正序数组的中位数

题目

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

题面

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

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

关键词

二分，维护指针

思路

目标就是找出第k个值，要做的就是从两个数组中去除比第k个值小的数，实现方法是通过移动指针，指针所指的左边都是可以去除的。

首先需要分奇偶讨论，获得第k小的值；

在getKth函数中，通过定义一个死循环：

① 先明确边界情况

对于其中一个数组已经完全“去除”的情况，则可以直接在另一个数组中找到；

对于k==1的情况，直接返回两个数组中指针所指的最小值即可；

② 然后对正常情况下的数组进行处理，通过二分的思想，比较两个数组中新的指针所指位置（注意判断边界）的大小，来确定哪部分是可以去除的，从而使k的值减去可以去除的数组大小，再移动指针，重复执行，直至到达边界情况。

代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int k = (m+n)/2;
        if((m+n)%2==0)
            return (getKth(nums1,nums2,k)+getKth(nums1,nums2,k+1))/2.0;
        else
            return getKth(nums1,nums2,k+1);

    }
    int getKth(vector<int>& nums1,vector<int>& nums2,int k){
        int m = nums1.size();
        int n = nums2.size();
        int idx1 = 0;
        int idx2 = 0;
        while(true){
            //边界情况
            if(idx1 == m)
                return nums2[idx2+k-1];
            if(idx2 == n)
                return nums1[idx1+k-1];
            if(k == 1)
                return min(nums1[idx1],nums2[idx2]);
            
            //正常情况
            int nidx1 = min(idx1+k/2,m)-1;
            int nidx2 = min(idx2+k/2,n)-1;
            int privot1 = nums1[nidx1];
            int privot2 = nums2[nidx2];
            if(privot1<=privot2){
                k -= nidx1-idx1+1;
                idx1 = nidx1+1;
            }
            else{
                k -= nidx2-idx2+1;
                idx2 = nidx2+1;
            }
            
        }
    }
};

总结

确实有些难，看了题解才勉强懂一点;

对于有序数组来说二分思想是非常重要的，而且对于二分之前也一直局限于移动左右指针，不会灵活运用。

参考

寻找两个有序数组的中位数 - 寻找两个正序数组的中位数 - 力扣（LeetCode）