题目:http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1105
题意:
数组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 2
1 2
2 3
3 4
Output示例
9
分析:
做这题根本没想到二分,一直在考虑优先队列什么的QAQ
以前做过的二分一般是二分下标,这题是二分值,然后找大于这个数的个数,然后根据数的个数于k比较在判断值是大了还是小了。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+10;
int n,k;
ll A[N], B[N];
ll check(ll x) { //找A[i]*B[j]>=x的数量
ll sum = 0;
for(int i = n; i >= 1; i--) {
int t = lower_bound(B + 1, B + 1 + n, (x - 1) / A[i] + 1) - B;
sum += n - t + 1;
if(sum == 0) break;
}
return sum;
}
int main() {
//freopen("in.txt","r",stdin);
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++) scanf("%lld%lld", &A[i], &B[i]);
sort(A + 1, A + 1 + n);
sort(B + 1, B + 1 + n);
ll l = A[1]*B[1], r = A[n] * B[n], m, ans;
while(l <= r) {
m = (l + r) >> 1;
ll f = check(m);
if(f < k) r = m - 1;
else l = m + 1,ans=m;
}
printf("%lld\n", ans);
return 0;
}