xdu 1068 Clinton’s beliefs

题目

两个数组，求所有乘积中的第k大的。

最多有10^10个乘积，直接算出来然后排序肯定是行不通的。

方法：二分。

分别对两个数组从小到大排序，然后二分乘积值x。

定义函数cal（num），求乘积大于num的个数。

则问题转化为求一个乘积x，使得大于x的乘积的个数>=k，并且使得大于等于x的乘积个数尽可能小（靠近k）。

求乘积大于num的个数的时候，有个技巧，顺序遍历a，逆序遍历b。设a的循环变量为i，b的循环变量为j，a中数的个数为n1，b中数的个数为n2

对于a[i]，遍历b的时候，加上条件：a[i]*b[j]>num，由于之前已经对b排好序，这样当a[i]*b[j]<=num的时候结束对b的遍历，统计n2-1-j即为b中和a[i]相乘大于num的乘积个数

对于b的循环变量j，一开始初始化为n2-1，每次遍历b的时候不需要再从头开始，因为a、b之前已经排好序，a[i]随着i的增长不断增大，b[j]不断减小，条件a[i]*b[j]>num对于之前的j一定成立。因此不必重置j的值。这样cal函数的时间复杂度为O(n1+n2)。算法总的时间复杂度为O((n1+n2)log(M*F))。M、F分别为a、b中数的最大值。

#include <stdio.h>
#include <algorithm>
#include <cmath>
#include <iostream>
using namespace std;
typedef long long LL;
#define N 10001
LL n1,n2, k, a[N], b[N];

LL cal(LL num)
{
	LL i, j, ans;
	if(a[n1-1]*b[n2-1]<=num) return 0;
	j = n2-1;
	ans = 0;
	for(i=0; i<n1; i++)
	{
		for(; j>=0 && a[i]*b[j]>num; j--);
		ans += n2-1-j;
	}
	return ans;
}

LL solve()
{
	LL l, r, mid, tmp;
	l=a[0]*b[0], r=a[n1-1]*b[n2-1];
	while(l<=r)
	{
		mid = (l+r)>>1;
		tmp = cal(mid);
		if(tmp>=k) l = mid+1;
		else r = mid-1;
	}
	return r+1;
}

int main()
{
	int i, j, t;
	scanf("%d", &t);
	while(t--)
	{
		scanf("%lld%lld%lld", &n1,&n2,&k);
		for(i=0; i<n1; i++) scanf("%lld", &a[i]);
		for(i=0; i<n2; i++) scanf("%lld", &b[i]);
		sort(a, a+n1);
		sort(b, b+n2);
		printf("%lld\n", solve());
	}
	return 0;
}