原题地址:http://acm.hdu.edu.cn/showproblem.php?pid=4334
说一下大概题意:给你5个数组,每个数组里有相同数量的元素,让你判断是否能在每个数组中选取1个元素,使得取完5个数后,相加为0
思路:
- 第一次就是直接写了五个for语句,然后意料之中超时了。
- 第二次想到了有个叫折半枚举的方法,于是就把5个数组分为了2和3,用一个新数组保存前两个数组相加的所有结果过,再用一个新数组保存后三个的结果,最后再枚举相加,然后还是超时了。
第三次就想到了可能是有过多的重复数据,于是多了一步去重操作,还是超时了(后话,这题数据相加结果中重复数据不是特别多)
第四次想到到了使用logn复杂度的二分搜索,枚举2个数组相加的结果,然后二分搜索3个数组相加结果中是否有符合题意的解。然后还是超时了。
然后想到了既然把5个数组分为2个和3个行不通,就只能再次细分为1 2 2.这样的话就用到了一个小技巧,然后就ac了,太艰难了。
事实上考虑如下问题,快速求解序列A,序列B中,是否有Ai+Bj=x ,记得是某知名公司的一道面试题吧,是两个指针的应用,将A,B升序排列,初试 i 指针指向A[1] ,j 指针指向 B[b_size] ,比较Ai + Bj 与 x 的大小。 若A[i]+B[j] < x , i 指针右移 ; 若 A[i]+B[j]>x , j 指针左移 。事实上是一个贪心的思想。(转自网络)
#include<stdio.h>
#include<iostream>
#include<algorithm>
typedef long long ll;
using namespace std;
ll a[5][205];
ll sum[40005];
ll sum1[40005];
int main()
{
int t,n;
scanf("%d",&t);
while(t--)
{
int flag=0;
int p=0,q=0;
scanf("%d",&n);
for(int i=0; i<5; i++)
{
for(int j=0; j<n; j++)
{
scanf("%I64d",&a[i][j]);
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<n; j++)
{
sum[p++]=a[0][i]+a[1][j];
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<n; j++)
{
sum1[q++]=a[2][i]+a[3][j];
}
}
sort(sum,sum+p);//记得要排序
sort(sum1,sum1+q);
for(int k=0; k<n; k++)
{
int i=0;//让sum数组从左到右
int j=p-1;//让sum1数组从右到左
while(j>=0&&i<=p-1)//判断结束条件
{
ll temp=sum[i]+sum1[j];
if(temp+a[4][k]<0)
{
i++;
}
else if(temp+a[4][k]>0)
{
j--;
}
else
{
flag=1;//flag为1表示找到了为0的值,就可以结束循环了
break;
}
}
if(flag) break;
}
if(flag) printf("Yes\n");
else printf("No\n");
}
return 0;
}