第K大的数

来源:互联网 发布:出差 准备 知乎 编辑:程序博客网 时间:2024/04/30 15:57
数组A和数组B,里面都有n个整数。数组C共有n^2个整数,分别是A[0] * B[0],A[0] * B[1] ......A[1] * B[0],A[1] * B[1]......A[n - 1] * B[n - 1](数组A同数组B的组合)。求数组C中第K大的数。
例如:A:1 2 3,B:2 3 4。A与B组合成的C包括2 3 4 4 6 8 6 9 12共9个数。
Input
第1行:2个数N和K,中间用空格分隔。N为数组的长度,K对应第K大的数。(2 <= N <= 50000,1 <= K <= 10^9)第2 - N + 1行:每行2个数,分别是A[i]和B[i]。(1 <= A[i],B[i] <= 10^9)
Output
输出第K大的数。
Input示例
3 21 22 33 4
Output示例

Output示例
9

AC代码:

#include<iostream>#include<algorithm>using namespace std;long long a[50010],b[50010];long long K,N;long long k,i,j;long long lef,righ,su;long long mid;long long cha(long long x)  {      long long ans;        ans=0;      j=N;   //这样可以剪掉没用的数据,节省时间,否则会超时    for(i=1;i<=N;i++){          for(;j>=0;j--){              if(a[i]*b[j]<=x)break;          }          ans+=j;      }      return ans;  }int main(){ios::sync_with_stdio(false); cin>>N>>K;k=N*N-K+1;for(i=1;i<=N;i++)cin>>a[i]>>b[i];sort(a+1,a+N+1);sort(b+1,b+N+1);su=0;lef=a[1]*b[1];righ=b[N]*a[N];while(righ-lef>1){mid=(lef+righ)>>1;if(cha(mid)>=k)righ=su=mid;else lef=mid;}cout<<su<<endl;return 0;}


0 0
原创粉丝点击