面试好题

来源:互联网 发布:电商大数据 下载 编辑:程序博客网 时间:2024/06/05 10:45

给定两个已经排好序的数组,让你求两个数组中第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;}



0 0