1540: 第k大数
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 665 Solved: 98
[Submit][Status][Web Board]
Description
有两个序列a,b,它们的长度分别为n和m,那么将两个序列中的元素对应相乘后得到的n*m个元素从大到小排列后的第k个元素是什么?
Input
输入的第一行为一个正整数T (T<=10),代表一共有T组测试数据。
每组测试数据的第一行有三个正整数n,m和k(1<=n, m<=100000,1<=k<=n*m),分别代表a序列的长度,b序列的长度,以及所求元素的下标。第二行为n个正整数代表序列a。第三行为m个正整数代表序列b。序列中所有元素的大小满足[1,100000]。
Output
对于每组测试数据,输出一行包含一个整数代表第k大的元素是多少。
Sample Input
3 3 2 3 1 2 3 1 2 2 2 1 1 1 1 1 2 2 4 1 1 1 1
Sample Output
3 1 1
HINT
Source
解析:不可能每个数去相乘然后排序。所有我们可以去二分查找。这里二分查找不是很好想到的。二分枚举一个数,求大于它的数有几个,若>k,说明值在大的地方,也就是右边,l=mid+1;反之亦然。这里还有一个地方,就是求大于二分枚举数的值有几个。我们可以贪心一下,排序。则l=a[0]*b[0], r=a[n-1]*b[m-1]。a从后往前,b从前往后。具体看代码f()函数
#include<bits/stdc++.h>
using namespace std;
#define e exp(1)
#define pi acos(-1)
#define mod 1000000007
#define inf 0x3f3f3f3f
#define ll long long
#define ull unsigned long long
#define mem(a,b) memset(a,b,sizeof(a))
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int n,m,k;
const int maxn=100005;
ll a[maxn],b[maxn];
ll f(ll x)
{
ll cnt=0;
int j=0;
for(int i=n-1; i>=0; --i)
{
for(; j<m; ++j)
{
if(a[i]*b[j]>=x)
{
cnt+=(m-j);
break;
}
}
}
return cnt;
}
int main()
{
int T;scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
for(int i=0; i<n; i++)scanf("%lld",&a[i]);
for(int i=0; i<m; i++)scanf("%lld",&b[i]);
sort(a,a+n);
sort(b,b+m);
ll l=a[0]*b[0];
ll r=a[n-1]*b[m-1];
ll ans=0;
while(l<=r)
{
ll mid=l+r>>1;
ll p=f(mid);
if(p>=k)l=mid+1,ans=mid;
else if(p<k)r=mid-1;
}
printf("%lld\n",ans);
}
return 0;
}