题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=4334
题目意思:
给5组数,每组含n个数(n<=200),在5组数中各选一个,问是否存在一种组合使这5个数的和为0。
解题思路:
这题数据很强,o(n^3*lgn过不了)
对于在两组数各选一个使得和为定值,有一种贪心的算法,时间复杂度为(n*lgn,实际上就是排序的算法时间复杂度)。
所以可以把第二组和第三组数合并组合40000个数作为第二组,第四组和第五组合并成40000个数作为第三组。
分别排序。
1、然后枚举第一组,初始时用p指针指向第二组中的第一个数,q指针指向第三组的最后一个数,以第二组为基准。
2、当三数大于0时,说明q指针后面的所有数与p指针后面的数肯定不行,大于当前的和值,同时说明p指针后面的只可能和q指针前面的凑,所以q指针往前移。
3、当三数小于0时,说明p与当前的q不行,p与q后面的也不行(由上一步得),p与q前面的更不行。所以p只能往后移。
贪心思想,所以最多只移动了40000+40000次。
所以总的时间复杂度为o(n^3).
代码:
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#define eps 1e-6
#define INF 0x1f1f1f1f
#define PI acos(-1.0)
#define ll __int64
#define lson l,m,(rt<<1)
#define rson m+1,r,(rt<<1)|1
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
/*
freopen("data.in","r",stdin);
freopen("data.out","w",stdout);
*/
ll s1[41000],s2[41000];
ll sa[6][210];
int cnt1,cnt2,n;
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(int i=1;i<=5;i++)
for(int j=1;j<=n;j++)
scanf("%I64d",&sa[i][j]);
//sort(sa[1]+1,sa[1]+n+1);
cnt1=cnt2=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
s1[++cnt1]=sa[2][i]+sa[3][j]; //把2、3组合并和把4、5组合并
s2[++cnt2]=sa[4][i]+sa[5][j];
}
sort(s1+1,s1+cnt1+1);
sort(s2+1,s2+cnt2+1);
int t=1; //离散化一下,去掉相同的
for(int i=2;i<=cnt1;i++)
if(s1[i]!=s1[i-1])
s1[++t]=s1[i];
cnt1=t;
t=1;
for(int i=2;i<=cnt2;i++)
if(s2[i]!=s2[i-1])
s2[++t]=s2[i];
cnt2=t;
/* if(sa[1][n]+s1[cnt1]+s2[cnt2]<0)
{
printf("Yes\n");
continue;
}*/
bool ans=false;
for(int i=1;i<=n&&!ans;i++)
{
//if(sa[1][i]+s1[1]+s2[1]>0)
// break;
for(int p=1,q=cnt2;p<=cnt1&&q>=1;) //贪心思想
{
ll tt=sa[1][i]+s1[p]+s2[q];
if(tt==0)
{
ans=true;
break;
}
else if(tt>0)
q--;
else
p++;
}
}
if(ans)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}