题意:
题目给了我们数组a和b,然后给了c数组中各个元素是由数组a和b相应元素相乘得到。让我们求出数组c中的第K大元素的值。
题解:
这道题目数组a和b的长度n是5e4,因此直接不能直接求出所有数组c元素的值。
这类求第K大的问题,有时可以使用二分的方法进行求解。
这道题目二分的思路就是,将数组a和数组b进行排序,首先二分这个第K大元素的值mid,然后在check函数当中,对数组a的每个元素二分找到与a[i]相乘大于mid的最小的b[j],那么显然b[j]之后的所有数组b的元素与a[i]相乘都会大于当前的mid,个数num累加n - j + 1,用num与K进行比较,即可知道当前第K大的数是大了还是小了,然后进一步二分即可。
细节见代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 5e4 + 10;
ll n, k;
ll a[MAXN], b[MAXN];
ll is(ll now, ll x) { // 二分与a[i]乘积大于now的最小的b[j]位置
ll l = 0, r = n - 1;
while (l < r) {
ll mid = l + r >> 1;
if (b[mid] * x > now) {
r = mid;
}
else {
l = mid + 1;
}
}
ll ans = n - l - 1;
if (b[l] * x > now) ans++;
return ans;
}
bool check(ll mid) {
ll num = 0;
for (int i = 0; i < n; i++) {
ll now = is(mid, a[i]);
num += now;
}
if (num + 1 <= k) return true;
return false;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
cin >> n >> k;
for (int i = 0; i < n; i++) {
cin >> a[i] >> b[i];
}
sort(a, a + n);
sort(b, b + n);
ll l = a[0] * b[0], r = a[n - 1] * b[n - 1];
// 二分第K大元素的值
while (l < r) {
ll mid = l + r >> 1;
if (check(mid)) {
r = mid;
}
else {
l = mid + 1;
}
}
cout << l << endl;
return 0;
}