题意:
现在给你三个数组,然后让你在每个数组中找一个数,使得a[i]+b[j]+c[k]=s成立。
思路:
肯定不能是for三次,而且for两次然后再二分这样子也是会T的。
所以就只for第一个数组,然后预处理出b[j]+c[k]的值,然后对s-a[i]这个值进行二分。
*注意一个细节,这里我没考虑到所以一开始WA了好几次。就是这里每个数组中的数是有可能是负数的,所以我一开始判x>(s-a[i])就continue这样是不对的。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<set>
#include<cmath>
using namespace std;
#define maxn 550
typedef __int64 ll;
ll d[500*500+50];
ll a[maxn],b[maxn],c[maxn];
ll l,n,m,k;
int bin(ll x,ll y,ll res){
ll mid;
ll ans=0;
while(x<y){
mid=x+(y-x)/2;
if(d[mid]<res) x=mid+1;
else if(d[mid]>res) y=mid;
else if(d[mid]==res) return 1;
}
return 0;
}
int solve(int x){
for(int i=1;i<=l;i++){
ll res=0;
res=x-a[i];
if(bin(0,k,res)) return 1;
}
return 0;
}
int main(){
int jj=1;
while(~scanf("%I64d%I64d%I64d",&l,&n,&m)){
for(int i=1;i<=l;i++) scanf("%I64d",&a[i]);
for(int i=1;i<=n;i++) scanf("%I64d",&b[i]);
for(int i=1;i<=m;i++) scanf("%I64d",&c[i]);
k=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
d[k++]=b[i]+c[j];
}
}
sort(d,d+k);
int T;
scanf("%d",&T);
printf("Case %d:\n",jj++);
while(T--){
int x;
scanf("%d",&x);
if(solve(x)){
printf("YES\n");
}
else{
printf("NO\n");
}
}
}
}