1105 第K大的数

1105 第K大的数

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

思路：开始没看清K的范围，不然可以用优先队列，这里用2个二分

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=50004;

ll n,m;
ll a[N],b[N];

ll hh(int y,ll x){
    int l=1,r=n,mid;
    int ans=1;
    while(l<=r){
            mid=(l+r)>>1;
        if(a[y]*b[mid]>=x){
            ans=mid;r=mid-1;
        }
        else l=mid+1;
    }
    return ans;
}
ll check(ll x){
    ll sum=0;
    for(int i=1;i<=n;i++){
        if(a[i]*b[n]<x) continue;
        sum+=n-hh(i,x)+1;
    }
    return sum;
}
int main(){
    scanf("%lld%lld",&n,&m);
    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]-1,r=a[n]*b[n]+1;;
    ll ans=0;
    while(l<=r){
        ll mid=(l+r)>>1;

        if(check(mid)>=m){
            l=mid+1;
            ans=mid;
        }
        else r=mid-1;
       
    }
    printf("%lld\n",ans);
}