1540: 第k大数
Time Limit: 10 Sec Memory Limit: 128 MB[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
【分析】
这道题虽然是10s,但是还是妥妥的超时....做过很多次这种题目了,总是不能一开始就想到可以二分查找..一定要有人提醒了一下或者是错了很久之后才发现,其实可以二分。
这道题的二分不难,大概就是可能一下子想不到吧...
显然这道题要先对a,b数组排序,这是一定的。
然后,l=a[0]*b[0],r=a[n-1]*b[m-1]开始去查找mid是第几大的数
除了二分之外,这里就算第几大的数这个思路直接关系到会不会TLE
模拟一下就可以发现,既然已经把a,b数组排序,那么当a[i]*b[j]>=mid的时候,下一个数只可能在a[i-1]*b[j](这里是因为我排序是升序)往后产生,所以这里的并不需要n*m的循环而是n+m,这两个速度显然不是一个数量级的,所以...就AC了~
另外有一点要注意,就是不能当mid返回的位置正好是k的时候就输出答案,因为mid这个位置正好是k的时候有两个情况,一个是mid就是我们要找的数,另外一个情况是mid刚好比这个序列中的k+1大的数大而比第k大的数小....好绕口...这个点是我wa两次之后才发现的...
【代码】
#include <stdio.h>
#include <algorithm>
using namespace std;
int n,m,k;
long long a[100000];
long long b[100000];
long long judge(long long mid)
{
int sum=0;
int j=0;
for(int i=n-1;i>=0;i--)
for (;j<m;j++)
if(a[i]*b[j]>=mid)
{
sum+=m-j;
break;
}
return sum;
}
int main()
{
int pp;scanf("%d",&pp);
while(pp--)
{
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<n;i++)scanf("%d",&a[i]);sort(a,a+n);
for(int i=0;i<m;i++)scanf("%d",&b[i]);sort(b,b+m);
long long l=a[0]*b[0];
long long r=a[n-1]*b[m-1];
long long ans;
while(l<=r)
{
long long mid=(r+l)/2;
int sum=judge(mid);
if(sum>=k)
ans=mid,l=mid+1;
else
r=mid-1;
}
printf("%lld\n",ans);
out:;
}
return 0;
}