第K小数

第K小数
( number .cpp/c/pas)

【问题描述】
有两个正整数数列，元素个数分别为N和M。从两个数列中分别任取一个数
相乘，这样一共可以得到N*M个数，询问这N*M个数中第K小数是多少。

【输入格式】
输入文件名为number.in。
输入文件包含三行。
第一行为三个正整数N,M和K。
第二行为N个正整数，表示第一个数列ai。
第三行为M个正整数，表述第二个数列bi。

【输出格式】
输出文件名为number.out。
输出文件包含一行，一个正整数表示第K小数。

【输入输出样例1】
2 3 4
1 2
2 1 3
3

【输入输出样例2】
5 5 18
7 2 3 5 8
3 1 3 2 5
16

【数据规模与约定】
n<=200000
m<=200000
k<=20000010000
ai,bi<=10^9

二分
利用两个指针检查

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<climits>
#include<string>
#include<cstdlib>
#include<ctime>
#include<map>
#define LL long long
#define MOD 1000000007
using namespace std;

LL n,m,k,i,a[200005],b[200005],l,r,mid;

LL check(LL x)
{
    LL p1=1,p2=m,sum=0;

    while(p1<=n&&p2>=1)
    {
        while(a[p1]*b[p2]>x&&p2>=1) p2--;       

        p1++;
        sum+=p2;
    }

    return sum;
}

void kth()
{
    l=a[1]*b[1];
    r=a[n]*b[m];

    while(l<=r)
    {
        mid=(l+r)>>1;

        if(check(mid)+1>k) r=mid-1;
        else l=mid+1;
    }

    printf("%lld",l);
}

int main()
{
    freopen("number.in","r",stdin);
    freopen("number.out","w",stdout);   

    scanf("%lld%lld%lld",&n,&m,&k);

    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);

    sort(a+1,a+n+1);

    for(i=1;i<=m;i++)
        scanf("%lld",&b[i]);

    sort(b+1,b+m+1);

    kth();

    return 0;
}