51nod-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

思路：对于 n<=5000,  用 n^2肯定超时，有题目要求 第 K 大的数，那么可以 对 答案二分查找，在 查找 答案 排在第几位时又可 用二分查找。

对于答案范围 l,r, 查找数 h=(l+r)/2，二分查找h排在第 p 位若 p>=K则说明 h比正确答案要小 l=h+1，反之说明要大，r=h+1.再二分查找 直到 l>r,则 r就是答案

Code:

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;

const int MAX_N=50005;
int n,m;
int a[MAX_N],b[MAX_N];

int Bound(int t,LL p);
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=0;i<n;++i)
		cin>>a[i]>>b[i];
	sort(a,a+n);
	sort(b,b+n);
	LL l=a[0]*b[0],r=(LL)a[n-1]*b[n-1];
	while(l<=r){
		int sum=0;
		LL h=(l+r)>>1;
		for(int i=n-1;i>=0&&sum<m;--i)
			sum+=Bound(i,h);
		if(sum>=m)	l=h+1;
		else	r=h-1;
	}
	cout<<r<<endl;
	return 0;
}

int Bound(int t,LL p)
{
	int l=0,h,r=n-1;
	while(l<=r){
		h=(l+r)>>1;
		if((LL)a[t]*b[h]>=p)	r=h-1;
		else	l=h+1;
	}
	return n-l;
}