LeetCode 4 寻找两个正序数组中的中位数(递归)

题目描述
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
示例 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
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000

题解:
暴力做法:两数组排序去中间值,时间复杂度O(nlogn)
递归做法:寻找中位数也就是把数组分成两部分,前一部分比中位数小,后一部分比中位数大,现在要寻找的是寻找第k小的数 其中k =nums1.size()+nums2.size() /2 。具体做法看该链接

注意:边界条件的先后顺序,首先需要看两个数组长度大小情况,该解法是规定数组1是短数组,数组2是长数组,其次的边界条件是nums1的起始位置已经到数组的末尾位置,最后一个边界条件是k==1,这样直接返回两个数组中最小值即可。

还有一种解法是二分,时间原因暂时没有看。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int total=nums1.size()+nums2.size();
        if(total%2==1) return find(nums1,0,nums2,0,total/2+1);
        int left=find(nums1,0,nums2,0,total/2);
        int right=find(nums1,0,nums2,0,total/2+1);
        return (left+right)/2.0;
    }
    int find(vector<int> &nums1,int i,vector<int> &nums2,int j,int k)
    {
       
        if(nums1.size()==i) return nums2[j+k-1];//如果第一个数组为空
        if(nums1.size()-i >nums2.size()-j) return find(nums2,j,nums1,i,k);//第一个数组个数大于第二个数组个数
        if(k==1) return min(nums1[i],nums2[j]);//数组中只有一个元素

        int si=min(int(nums1.size()),i+k/2),sj=j+k/2;
        if(nums1[si-1]>nums2[sj-1]) return find(nums1,i,nums2,sj,k-(sj-j));
        else return find(nums1,si,nums2,j,k-(si-i));
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值