4. 寻找两个正序数组的中位数(hard)

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

进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?

 

示例 1:

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

二分法查找两个正序排列的数组的中位数,设中位数为K,那么先在数组1找K/2位置,数组2找K/2位置,根据比较结果缩小范围

class Solution {
private:
//i,j分别是nums1,nums2的起始位置
    int findK(vector<int> nums1, int i, vector<int> nums2, int j, int k)
    {
        // nums1为空数组
        if(i >= nums1.size()) return nums2[j + k - 1];
        // nums2为空数组
        if(j >=  nums2.size()) return nums1[i + k - 1];
        if(k == 1) return min(nums1[i], nums2[j]);

        int mid1 = INT_MAX, mid2 = INT_MAX;
        if(i + k / 2 - 1 < nums1.size())
            mid1 = nums1[i + k / 2 - 1];
        if(j + k / 2 - 1 < nums2.size())
            mid2 = nums2[j + k / 2 - 1];
        if(mid1 < mid2)
            return findK(nums1, i + k / 2, nums2, j, k - k / 2);
        else
            return findK(nums1, i, nums2, j + k / 2, k - k / 2);

    }
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size();
        int len2 = nums2.size();
        int left = (len1 + len2 + 1) / 2;
        int right = (len1 + len2 + 2) / 2;
        return (findK(nums1, 0, nums2, 0, left) + findK(nums1, 0, nums2, 0, right)) / 2.0;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值