HDU 2141 can you find it [二分]

1.题意：给出三组数，给出一个数X，试问是否能从三个数组中各选一个数，A,B,C使得X=A+B+C

2.分析：题设的输入数据为先是依次给出三组数的长度L、N、M，下面三行给出三组数的内容，再下面一行给出要查询的数X的个数S，最后S行是依次给出S个查询值X。其中1<=L, N, M<=500, 1<=S<=1000。首先若是朴素的考虑三个数组的任意组合，那么有L*N*M种组合，结合数据规模O(N*3)看太大了，所以将上式变形为A+B=X-C，考虑前两个数组的任何组合，有L*N种,这样对于每一个X,等式右边就有M种情况，遍历这M种情况，对于右边每一种取值，二分寻找左边L*N种取值，这样就能在O(NlogN)的规模下求解。

3.代码：Tips:数据中包含负数，且注意运算过程中有可能超过int的范围

# include <iostream>
# include <cstdio>
# include <algorithm>
using namespace std;
const int MAXN=505;
int COUNT=0;
int L,N,M,S;
long long A[MAXN],B[MAXN],C[MAXN];
long AB[MAXN*MAXN],snum[2*MAXN];
void Init()
{
    for(int i=0;i<L;i++)
        scanf("%lld",&A[i]);
    for(int i=0;i<N;i++)
        scanf("%lld",&B[i]);
    for(int i=0;i<M;i++)
        scanf("%lld",&C[i]);
    scanf("%d",&S);
    int temp=0;
    for(int i=0;i<L;i++)
        for(int j=0;j<N;j++)
            AB[temp++]=A[i]+B[j];
    sort(C,C+M);
    sort(AB,AB+L*N);
    for(int i=0;i<S;i++)
        scanf("%lld",&snum[i]);
}
bool judge(long long n)
{
    for(int i=0;i<M;i++)
    {
        int l=0;
        int r=N*L-1;
        if((C[i]<n-AB[r])||(C[i]>n-AB[l])) continue;//不用longlong就会WA 
        while(l<=r)
        {
            int mid=l+(r-l)/2;
            if(AB[mid]+C[i]==n) return true;
            else if(AB[mid]+C[i]>n) r=mid-1;
            else l=mid+1; 
        }
    }    
    return false;
}
void Solve()
{
    printf("Case %d:\n",++COUNT);
    for(int i=0;i<S;i++)
    {
        long long temp=snum[i];
        if(judge(temp)) printf("YES\n");
        else printf("NO\n");
    }
}
int main()
{
    while(scanf("%d%d%d",&L,&N,&M)!=EOF)
    {
        Init();
        Solve();
    }
    return 0;
}