笔记LeetCode 4.寻找两个正序数组的中位数

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

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

输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5

 注:1.中位数点:(m+n+1)/2 和 (m+n+2)/2 可以同时满足 两个数组长度之和 为奇数和偶数的情况

         2.处理边界情况,当其中一个数组的长度不够时,可以得出中位数一定在另一数组当中

         3.二分法,每次递归时,由于两个数组都是有序数组,所以每次递归通过判断两个数组的中点大小,删掉小的数组一半,即下次递归起始位置为上次数组的一半后开始。

         4.去中位点时要注意,当数组长度不够的时候,将其置为INT_MAX

class Solution {
public:
    int findKth(vector<int>& nums1,int i,vector<int>& nums2,int j,int k){
        //i:nums1的起始位置,j:nums2的起始位置,k:目标位置
        //边界问题,当某一数组的长度不足时,说明中位数一定在另一数组,可以通过坐标直接返回中位数
        if(i==nums1.size()) return nums2[j+k-1]; //nums1为空数组
        if(j==nums2.size()) return nums1[i+k-1]; //nums2为空数组
        if(k==1){
            return min(nums1[i],nums2[j]);
        }
        int midVal1 = (i + k/2 -1 < nums1.size()) ? nums1[i + k/2 -1] : INT_MAX;
        int midVal2 = (j + k/2 -1 < nums2.size()) ? nums2[j + k/2 -1] : INT_MAX;
        if(midVal1<midVal2){
            return findKth(nums1,i+k/2,nums2,j,k-k/2);
        }else{
            return findKth(nums1,i,nums2,j+k/2,k-k/2);
        }
    }
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int left = (m+n+1)/2;
        int right = (m+n+2)/2;
        return (findKth(nums1,0,nums2,0,left) + findKth(nums1,0,nums2,0,right)) / 2.0;
        
    }
};

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值