题意:能不能从给定的三个序列中找出满足ai+bj+ck==给定的x值,能则输出YES,否则输出NO。
思路:暴搜绝对会超时,由于区间长度为静态,考虑排序后二分(二分的前提一定是有序)。二分有两种选择,其中一种会超时:1.在c中二分查找 x-a[i]-b[j] 是否存在,那么时间复杂度为n * m * log( p )(设a,b,c数组的长度为n,m,p),会超时;
2.还有一种更好的方法,先预处理出a,b,数组中所有可能组合情况的和(放入ab数组),在ab中二分查找 x-c[i]是否存在,时间复杂度为p * log ( n * m);
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
//朴素暴搜时间复杂度为n^3会TIE;
//对于区间为静态(即无增删操作时),考虑二分查找;
//这里二分可以总值减去前两个值去二分第三个值,但是这样时间复杂度有n*m*logp,查询有k次,
//这样做还是会TLE(为什么我知道,因为我这样做TLE了...)
//故二分应该预处理出前两项,用总值减去第三项,取预处理出来的数组中二分,
//这样复杂度变为了p*log(n*m),还要注意类型要开long long;
ll a[666],b[666],c[666],ab[666*666];//ab数组用于预处理a,b两数组的和(数组要开得足够大);
int main()
{
int i,j;
int n,m,p;
int t,k=0;
ll x;
while(scanf("%d%d%d",&n,&m,&p)!=EOF)
{
for(i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
for(i=1;i<=m;i++)
{
scanf("%lld",&b[i]);
}
for(i=1;i<=p;i++)
{
scanf("%lld",&c[i]);
}
int tot=0;
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
ab[++tot]=a[i]+b[j];//预处理出a,b两数组的和;
}
}
sort(ab+1,ab+1+tot);//二分的条件为序列有序,故一定要先排序;
sort(c+1,c+1+p);
scanf("%d",&t);//t次询问;
printf("Case %d:\n",++k);
while(t--)
{
scanf("%lld",&x);
if((x<ab[1]+c[1])||(x>ab[tot]+c[p]))//若x比最小的还小或者x比最大的还大就不用搜了
{
printf("NO\n");
continue;
}
bool flag=true;//flag==false时为找到了满足条件的数;
for(i=1;i<=p&&flag;i++)
{
ll ans=x-c[i];
int l=1;
int r=tot;//对预处理出来的ab数组进行二分;
while(l<=r)
{
int mid=(l+r)>>1;
if(ab[mid]==ans)
{
flag=false;//找到符合条件的解;
break;
}
else
if(ab[mid]<ans)
{
l=mid+1;
}
else
{
r=mid-1;
}
}
}
if(!flag)
{
printf("YES\n");
}
else
{
printf("NO\n");
}
}
}
return 0;
}