ZCMU 1540: 第k大数(二分答案)

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

3 3 2 3 1 2 3 1 2 2 2 1 1 1 1 1 2 2 4 1 1 1 1

Sample Output

3 1 1

对两个数组进行排序，并在最小值和最大值中进行二分，check函数中采取双指针来进行操作。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=100001;
ll a[maxn];
ll b[maxn];
ll n,m,k;
bool cmp(int a,int b)
{
	return a>b;
}
ll check(ll x)
{
    int j=m-1;
    ll ans=0;
    for(int i=0;i<n;i++)
    {
        while(j>=0&&(ll)a[i]*b[j]<x)
            j--;
        ans+=j+1;
    }
    return ans;
}
int main()
{
	std::ios::sync_with_stdio(0);
	std::cin.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m>>k;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    for(int i=0;i<m;i++)
    {
        cin>>b[i];
    }
    sort(a,a+n,cmp);
    sort(b,b+m,cmp);
    ll cmd=a[n-1]*b[m-1];
    ll r=a[0]*b[0],l=cmd;
    ll mid;
    while (l<r) {
        mid=l+r+1>> 1;
        if (check(mid)>=k)l=mid;
        else r = mid-1;
    }
    cout<<l<<endl;
}
    return 0;
}