给定两个已经排好序的数组,让你求两个数组中第K大的数字,如何实现?
方法一:暴力,时间复杂度为O(K)。
方法二:二分,时间复杂度O(log(N+M)),N和M为两个数组的长度。每次二分舍掉一个数组的一半长度。
看代码再去想思路吧,也许容易些~
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
using namespace std;
typedef long long ll;
int Find(int a[],int la,int ra,int b[],int lb,int rb,int k)
{
//cout<<"a: "<<la<<' '<<ra<<" b: "<<lb<<' '<<rb<<endl;
int mida=(la+ra)/2,midb=(lb+rb)/2;
if(la>ra)
return b[lb+k-1];
if(lb>rb)
return a[la+k-1];
if(a[mida]<b[midb])
{
if(mida-la+1+midb-lb+1>k)
return Find(a,la,ra,b,lb,midb-1,k);
else
return Find(a,mida+1,ra,b,lb,rb,k-(mida-la+1));
}
else
{
if(mida-la+1+midb-lb+1>k)
return Find(a,la,mida-1,b,lb,rb,k);
else
return Find(a,la,ra,b,midb+1,rb,k-(midb-lb+1));
}
}
int main()
{
int a[5]= {1,2,4,8,10},b[4]= {2,3,4,5};
cout<<Find(a,0,4,b,0,3,4)<<endl;
return 0;
}