1540: 第k大数
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 573 Solved: 73
[ Submit][ Status][ Web Board]
Description
有两个序列a,b,它们的长度分别为n和m,那么将两个序列中的元素对应相乘后得到的n*m个元素从大到小排列后的第k个元素是什么?
Input
输入的第一行为一个正整数T (T<=10),代表一共有T组测试数据。
每组测试数据的第一行有三个正整数n,m和k(1<=n, m<=100000,1<=k<=n*m),分别代表a序列的长度,b序列的长度,以及所求元素的下标。第二行为n个正整数代表序列a。第三行为m个正整数代表序列b。序列中所有元素的大小满足[1,100000]。
Output
对于每组测试数据,输出一行包含一个整数代表第k大的元素是多少。
Sample Input
33 2 31 2 31 22 2 11 11 12 2 41 11 1
Sample Output
311
HINT
Source
具体解析,看下代码
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <iostream>
#include <string>
#define ll long long
using namespace std;
const int maxn=1000000+5;
ll a[maxn],b[maxn];
ll t,n,m,k;
bool cmp(int A,int B){
return A > B;
}
ll jdg(ll mid){
ll cnt=0;
ll j=1;
for(ll i=n;i>=1;i--){
/*
这里也很细节,对a数组从最大的开始,对b数组从最小的开始,如果a*b里面某个比较小的了,
那么a[i]后面的那个题,肯定乘以比当前b[j]大的数,然后就优化出来了。
*/
while(j <=m){
if(a[i]*b[j] >= mid){
cnt+=m-j+1;
break ;
}
else{
j++;
}
}
}
return cnt;
}
int main(){
scanf("%lld",&t);
while(t--){
scanf("%lld%lld%lld",&n,&m,&k);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int j=1;j<=m;j++) scanf("%lld",&b[j]);
sort(a+1,a+1+n);
sort(b+1,b+1+m);
ll mid,left,right;
left=a[1]*b[1];/*最小值*/
right=a[n]*b[m];/*最大值*/
ll ans;
while(left <= right){
mid=(left+right)/2;
/*这里可以是一点点逼近吧*/
ll ce=jdg(mid);
if(ce >= k){
ans=mid;
left=mid+1;
}
else{
right=mid-1;
}
}
printf("%lld\n",ans);
}
return 0;
}