【杭电】[2141]Can you find it?

因为N最大为500
所以如果开三重循环进行判断的话会超时
那么可以思考如何进行优化

A_i+B_j+C_k=X
⇒
A_i+B_j=X-C_k

所以可以预处理A_i+B_j
保存之后进行排序
然后对于每个询问
循环C然后在预处理中
二分查找有无X-C_k

#include<stdio.h>
#include<algorithm>
using namespace std;
int l,n,m,cnt;
int a[520],b[520],c[520],res[520*520];
bool find(int x) {
    int l=0,r=cnt-1;
    while(l<=r) {
        int mid=(l+r)/2;
        if(res[mid]>x)
            r=mid-1;
        else if(res[mid]<x)
            l=mid+1;
        else
            return true;
    }
    return false;
}
int main() {
    int kase=0;
    while(scanf("%d %d %d",&l,&n,&m)!=EOF) {
        printf("Case %d:\n",++kase);
        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]);
        cnt=0;
        for(int i=0; i<l; i++)
            for(int j=0; j<n; j++)
                res[cnt++]=a[i]+b[j];
        sort(res,res+cnt);
        int T;
        scanf("%d",&T);
        while(T--) {
            int t;
            scanf("%d",&t);
            int i;
            for(i=0; i<m; i++) {
                if(find(t-c[i]))
                    break;
            }
            if(i<m)
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}

题目地址:【杭电】[2141]Can you find it?