两个有序数组中位数

等价于有序数组第K小

首先我们可以不断分割得到这个 k k k
首先两个数组都分配 k 2 \frac{k}{2} 2k,如果第 k 2 \frac{k}{2} 2k个数,第一个数组小于第二个数组,那么第一个数组的 k 2 \frac{k}{2} 2k个数一定不是答案。
问题可以变成一个子问题,上一个数组从 k 2 + 1 \frac{k}{2}+1 2k+1开始。

相当于我们每次要去掉 k / 2 k/2 k/2个不可能的元素。
如果有某个数组到底了,没有 k / 2 k/2 k/2个元素了,那么肯定是从另一个数组扣除。
为什么呢?
我们考虑如果答案在上面,下面至少要有 > k / 2 >k/2 >k/2个去掉。
如果答案在下面,也要去掉至少 k > 2 k>2 k>2个。
因为上面是 < k / 2 <k/2 <k/2的,所以下面无论如何第 k k k个也比 k / 2 k/2 k/2多。

最后递归得到结果。

class Solution {
public: 
    int findK(vector<int>& nums1,int i,vector<int>& nums2,int j,int k){
        if(i >= nums1.size())return nums2[j+k-1];
        if(j >= nums2.size())return nums1[i+k-1];//数组为空
        if(k == 1)return min(nums1[i],nums2[j]);//结束条件
        int midl = i+k/2-1 < nums1.size() ? nums1[i+k/2-1] : 0x3f3f3f3f;
        int midr = j+k/2-1 < nums2.size() ? nums2[j+k/2-1] : 0x3f3f3f3f;
        if(midl <= midr)return findK(nums1,i+k/2,nums2,j,k-k/2);//如果前者小,那肯定不要前者的一半
        else return findK(nums1,i,nums2,j+k/2,k-k/2);//如果后者小,肯定不要后者
    }//算法核心在于二分区间,对于k然后不断缩小范围
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n = nums1.size(),m = nums2.size();
        int l = findK(nums1,0,nums2,0,(n+m+1)/2);
        int r = findK(nums1,0,nums2,0,(n+m+2)/2);
        return (l + r) / 2.0;
    }
};

非递归

class Solution {
public: 
    int findK(vector<int>& nums1,vector<int>& nums2,int k){
        int i = 0,j = 0;
        while(true){
            if(i >= nums1.size())return nums2[j+k-1];
            if(j >= nums2.size())return nums1[i+k-1];//数组为空
            if(k == 1)return min(nums1[i],nums2[j]);//结束条件
            int midl = i+k/2-1 < nums1.size() ? nums1[i+k/2-1] : 0x3f3f3f3f;
            int midr = j+k/2-1 < nums2.size() ? nums2[j+k/2-1] : 0x3f3f3f3f;
            if(midl <= midr)i += k/2;
            else j += k/2;
            k -= k/2;
        }
    }//算法核心在于二分区间,对于k然后不断缩小范围
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n = nums1.size(),m = nums2.size();
        int l = findK(nums1,nums2,(n+m+1)/2);
        int r = findK(nums1,nums2,(n+m+2)/2);
        return (l + r) / 2.0;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值