1105 第K大的数

1105 第K大的数 

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

数组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

    思路:只要我们在所在答案的范围内找到小于答案的个数cnt，然后用k去和cnt比较(这里是求第k大  我们把数组从小到大排序 把k转换为k大前有多少个数比k小 用kt = n*n-k+1来表示 )

    代码:

    

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 5e+5;
ll a[maxn],b[maxn];
int main(){

    ll n,k;
    while(cin>>n>>k){
        for(int i=1;i<=n;i++){
            cin>>a[i]>>b[i];
        }
        sort(a+1,a+n+1);
        sort(b+1,b+n+1);
        k = n*n-k+1;
        ll l=a[1]*b[1];
        ll r=a[n]*b[n];
        ll ans=0,cnt;
        while(l+1<r){
            cnt = 0;
            ll mid = (l+r)>>1;
           // cout<<mid<<endl;
            ll j = n;
            for(ll i=1;i<=n;i++){
                for(;j>=0;){
                    if(a[i]*b[j]>mid) j--;
                    else{
                        break;
                    }
                }
                cnt+=j;
            }

            if(cnt>=k){
                ans = mid;
                //cout<<mid<<endl;
                r = mid;
            }
            else{
                l=mid;
            }

        }
        cout<<ans<<endl;
    }

    return 0;
}