HDU - 2141：Can you find it?（二分搜索）

題目鏈接：HDU - 2141：Can you find it？

題意：給你三組數，要你各從三組數中挑出一個，滿足Ai+Bj+Ck=X$A_i+B_j+C_k=X$ ，若滿足輸出YES，否則NO

思路：典型的二分搜索，先A、B求和並排序，然後二分搜索ab=x-c

AC代碼：

#include<iostream>
#include<cstdio>
#include<algorithm>
#define maxn 510
using namespace std;
int a[maxn],b[maxn],c[maxn],ab[maxn*maxn];
int L,N,M,k;

bool judge(int x)
{
    int left=0,right=k;
    while(left<right)
    {
        int mid=left+(right-left)/2;
        if(ab[mid]<x) left=mid+1;
        else if(ab[mid]>x) right=mid;
        else return true;
    }
    return false;
}

int main()
{
    int cas=1;
    while(~scanf("%d%d%d",&L,&N,&M))
    {
        printf("Case %d:\n",cas++);
        for(int i=0; i<L; i++)
            scanf("%d",&a[i]);
        for(int i=0; i<N; i++)
            scanf("%d",&b[i]);
        for(int i=0; i<M; i++)
            scanf("%d",&c[i]);
        k=0;
        for(int i=0; i<L; i++)
            for(int j=0; j<N; j++)
                ab[k++]=a[i]+b[j];
        sort(ab,ab+k);
        int s,x;
        scanf("%d",&s);
        while(s--)
        {
            int flag=0;
            scanf("%d",&x);
            for(int j=0; j<M; j++)
            {
                int y=x-c[j];
                if(judge(y)) {flag=1;break;}
            }
            if(flag) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}