中石油4875 第k大数（二分）

4875: 第k大数

时间限制: 10 Sec  内存限制: 128 MB
提交: 80  解决: 28
[ 提交][ 状态][ 讨论版]

题目描述

有两个序列a，b，它们的长度分别为n和m，那么将两个序列中的元素对应相乘后得到的n*m个元素从大到小排列后的第k个元素是什么？

输入

输入的第一行为一个正整数T (T<=10)，代表一共有T组测试数据。

每组测试数据的第一行有三个正整数n，m和k(1<=n, m<=100000,1<=k<=n*m)，分别代表a序列的长度，b序列的长度，以及所求元素的下标。第二行为n个正整数代表序列a。第三行为m个正整数代表序列b。序列中所有元素的大小满足[1,100000]。

输出

对于每组测试数据，输出一行包含一个整数代表第k大的元素是多少。

样例输入

33 2 31 2 31 22 2 11 11 12 2 41 11 1

样例输出

311

题解：暴力的话，10s也会超时，想到用二分，二分的区间是谁，因为a和b序列之间的数字是无序的相乘也是无序，但是我们可以对a和b序列产生的最小值到最大值查找，进行二分，判断比这个数大的数字有多少个与k比较，但是当找到一个数，比它大的数字的个数正好为k个的话，要考虑当前的这个数字，是不是a和b序列产生的，可能这个数字不存在。

每次二分都想不到，只能多做多想了。

代码：

#include <stdio.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5+7;
ll a[N],b[N];
int n, m;
int cmp(int c, int d) {
    return c > d;
}

ll judge(ll mid) {
    ll sum = 0;
    int j = m-1;
    for(int i = 0; i < n; i++)
        for(; j >=0; j--)
            if(a[i]*b[j] >= mid) {
                sum += j+1;
                break;
            }

    return sum;
}

int main() {
    int t;
    ll k;
    scanf("%d", &t);

    while(t--) {
        scanf("%d%d%lld", &n, &m, &k);
        for(int i = 0; i < n; i++)
            scanf("%lld", &a[i]);
        for(int i = 0; i < m; i++)
            scanf("%lld", &b[i]);
        sort(a, a+n, cmp);
        sort(b, b+m, cmp);

        ll l = a[n-1]*b[m-1];
        ll r = a[0]*b[0];

        while(l<=r) {
            ll mid = (l+r)/2;
            if(judge(mid) >= k)
                l = mid+1;
            else
                r = mid-1;
        }

        printf("%lld\n", l-1);
    }
    return 0;
}