题目链接:https://acm.zcmu.edu.cn/JudgeOnline/problem.php?id=1540
题目大意
t组数据,每组数组 两个数组a,b长度分别数n,m,现在要他们两两相乘得到n*m个数,问这些数中第k大数是多少?
思路
n,m范围很大1e5,直接找O(n*m)还是会超时,可以考虑二分这个答案。
将a,b数组按从小到大排序,那么二分左边界是a[1]* b[1],二分右边界是a[n]*b[n], 判断条件就是比mid大的数的个数不低于k。
ac代码
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e6 + 5;
int a[maxn], b[maxn], n, m;
ll k;
bool check(ll x){
int j = 1, sum = 0;
for(int i = n; i >= 1; i --){ //这里a降序遍历,如果a[i]*b[j]>=x,那么a[i+1]*b[j]肯定<x,a[i+1]*b[j-1]也肯定<x,因此这里的j没必要从头开始了
for(; j <= m; j ++){
if(1ll * a[i] * b[j] >= x){ //由于b是升序的,如果这里就大于等于x了,那么后面的所有j肯定都满足了,直接记录数量
sum += m - j + 1;
break;
}
}
}
return sum >= k;
}
int main(){
int t; scanf("%d",&t);
while(t --){
scanf("%d%d%lld",&n,&m,&k);
for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
for(int i = 1; i <= m; i ++) scanf("%d", &b[i]);
sort(a + 1, a + n + 1);
sort(b + 1, b + m + 1);
ll l = 1ll * a[1] * b[1], r = 1ll * a[n] * b[m], ans = -1;
while(l <= r){
ll mid = l + r >> 1;
if(check(mid)) l = mid + 1, ans = mid;
else r = mid - 1;
}
printf("%lld\n", ans);
}
return 0;
}