【杭电2141】Can you find it?_杭电 2141题-优快云博客

本文链接：https://blog.youkuaiyun.com/better_space/article/details/52039439

本文分享了一种解决ACM竞赛中特定问题的有效算法优化方案。通过对三个输入数组的组合匹配，采用二分查找和排序策略减少计算复杂度，显著提高了程序运行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里写图片描述

我首先想到的是三重循环—-超时，后来想前两个数组双重循环套第三个数组二分查找—-还是超时，再后来就是现在的代码思路了300多ms。

其实和第二种想法很接近了，就是二分的对象变了，对第三个数组遍历循环，对前两个数组各项求和得到的数组求二分，因为这样就大大大大减少了循环次数，二分是省时间的嘛。

本题难点算法优化。

#include<cmath>
#include<algorithm>
#include<stdio.h>
using namespace std;
int a[505],b[505],c[505];
int dp[250025];
int main() {
    int i,j,k,m,n,s,t,cas=0,p,flag;
    while(scanf("%d%d%d",&k,&m,&n)!=EOF) {
        for(i=0; i<k; i++)
            scanf("%d",&a[i]);
        for(i=0; i<m; i++)
            scanf("%d",&b[i]);
        for(i=0; i<n; i++)
            scanf("%d",&c[i]);
        p=0;
        for(i=0; i<k; i++)
            for(j=0; j<m; j++)
                dp[p++]=a[i]+b[j];
        sort(dp,dp+p);
        printf("Case %d:\n",++cas);
        scanf("%d",&t);
        while(t--) {
            scanf("%d",&s);
            flag=0;
            for(i=0; i<n; i++) {
                int l=0,r=p-1,mid;
                while(l<=r) {
                    mid=(l+r)/2;
                    if(dp[mid]+c[i]==s) {
                        flag=1;
                        break;
                    } else if(dp[mid]+c[i]<s)
                        l=mid+1;
                    else if(dp[mid]+c[i]>s)
                        r=mid-1;
                }
                if(flag)
                    break;
            }
            if(flag)
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}