LeetCode Median of Two Sorted Arrays(对两个数组同时进行二分搜索)

https://leetcode.com/problems/median-of-two-sorted-arrays/

题意:给定两个排好序的数组,问这两个数组组合成一个新数组后,中位数是多少?要求复杂度必须为log(n+m)(n,m分别表示两个数组的长度)。

思路:log级别的复杂度只能二分了。

可以这样考虑。需要查找这两个数组中第(n+m)/2大的数是谁,那么我在第一个数组里找第p大,在第二个数组里找第(n+m)/2-p大的数,如果这两个数刚好相等,说明第(n+m)/2大的数就是这个。

如果不相等,那么值小的那个数组的那一部分可以直接舍弃。

具体分析参考这个:https://www.jianshu.com/p/9bd57fd52062  他写的非常详细。

代码:
 

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        if((nums1.size()+nums2.size())&1)
        {
            int l1=nums1.size(),l2=nums2.size();
            return dfs(nums1,nums2,l1,0,l2,0,(l1+l2)/2+1);
        }
        else
        {
            int l1=nums1.size(),l2=nums2.size();
            return 1.0*(dfs(nums1,nums2,l1,0,l2,0,(l1+l2)/2)+dfs(nums1,nums2,l1,0,l2,0,(l1+l2)/2+1))/2;
        }
    }
    double dfs(vector<int> &num1,vector<int> &num2,int len1,int start1,int len2,int start2,int k)
    {
        if(len1>len2)
        {
            return dfs(num2,num1,len2,start2,len1,start1,k);
        }
        if(len1==0)
        {
            return num2[start2+k-1];
        }
        int p=min(k/2,len1);
        int q=k-p;
        if(k==1)
        {
            return min(num1[start1],num2[start2]);
        }
        if(num1[start1+p-1]==num2[start2+q-1])
        {
            return num1[start1+p-1];
        }
        else if(num1[start1+p-1]>num2[start2+q-1])
        {
            return dfs(num1,num2,len1,start1,len2-q,start2+q,k-q);
        }
        else
        {
            return dfs(num1,num2,len1-p,start1+p,len2,start2,k-p);
        }
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值