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);
}
}
};