题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2141
这个题目看起来好像直接可以暴力,因为数据量好像不大,但是一旦暴力几个乘起来就大了
其实这个题目可以二分枚举,枚举出答案
我是把上面的式子稍微变下形:A+B+C=X->A+B=X-C
那么就分成AB和XC两组,两个枚举其中一个,在另外一个几集合里面找答案
如果不变形就在三个和一个里面找可能会超时间,两两组合时间省好多!
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define maxn 1000000
//int po[maxn];
int f[1000],s[1000],t[1000],ans[1000];
int first[maxn];
int a,b,c,d,n;
bool binary(int l,int r,int q,int *p)
{
int mid;
while(l<=r)
{
mid=(l+r)>>1;
if(p[mid]>q)
r=mid-1;
else if(p[mid]<q)
l=mid+1;
else
return true;
}
return false;
}
int init()
{
int i,j,k=0,pos=0,len;
for(i=0;i<a;i++)
for(j=0;j<b;j++)
first[pos++]=f[i]+s[j];
len=pos;
sort(first,first+len);
// printf("n:%d\n",n);
for(i=0;i<n;i++)
{
for(j=0;j<c;j++)
if(binary(0,len-1,ans[i]-t[j],first))
{
printf("YES\n");
break;
}
if(j==c)
printf("NO\n");
}
return 0;
}
int main()
{
int i,j,k=0;
while(scanf("%d%d%d",&a,&b,&c)!=EOF)
{
for(i=0;i<a;i++)
scanf("%d",&f[i]);
for(i=0;i<b;i++)
scanf("%d",&s[i]);
for(i=0;i<c;i++)
scanf("%d",&t[i]);
scanf("%d",&n);
// printf("haha:%d\n",n);
for(i=0;i<n;i++)
scanf("%d",&ans[i]);
printf("Case %d:\n",++k);
init();
}
return 0;
}