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

给定两个大小分别为 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

提示:

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

解题思路:
二分查找

  1. 两个数组找第k个元素,a数组找第k/2个元素,b数组k/2个元素,如果a的大,那么b的前k/2个元素一定不包括第k个元素。如果相等,那么这就是第k个元素。
  2. 注意边界情况 如果其中一个数组为0,另一个数组的第k个就是答案,如果k==1,那么去两个数组当前第一个元素最小值
    代码如下:
class Solution {
public:

    //中位数 偶数 (m + n) /2 , (m + n) / 2 + 1
    //nums1 取 k/2 ,nums2取 k/2 比较大小,如果一边已经不足,那么另一边第k个就是答案
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int k = nums1.size() + nums2.size();
        int m = nums1.size();
        int n = nums2.size();
        if (k % 2 == 0) return (helper(nums1, 0, m, nums2, 0, n, k/2) + helper(nums1, 0, m, nums2, 0, n, k/2 +1) ) / 2.0;
        else return helper(nums1, 0, m, nums2, 0, n, k/2 + 1);
    }
    double helper(vector<int>& nums1, int l1, int count1, vector<int>& nums2, int l2, int count2, int k) {
        if (count2 < count1) return helper(nums2, l2, count2, nums1, l1, count1, k);
        if (count1 == 0) return nums2[l2 + k -1];
        if (k == 1) return min(nums1[l1], nums2[l2]);
        int aoffset = min(count1, k / 2);
        int boffset = k - aoffset;
        if (nums1[l1 + aoffset - 1] < nums2[l2 + boffset - 1]) return helper(nums1, l1 + aoffset, count1 - aoffset, nums2, l2, count2, k - aoffset );
        else if (nums1[l1 + aoffset - 1] > nums2[l2 + boffset - 1]) return helper(nums1,l1, count1, nums2, l2 + boffset, count2 - boffset, k - boffset);
        else return  nums1[l1 + aoffset -1];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值