【杭电2141】Can you find it?

系统管理员 2022-09-24 05:29 177阅读 0赞

![这里写图片描述][20160726203553997]  
![这里写图片描述][20160726203601893]

我首先想到的是三重循环—-超时，后来想前两个数组双重循环套第三个数组二分查找—-还是超时，再后来就是现在的代码思路了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;
    }

[http://acm.hdu.edu.cn/showproblem.php?pid=2141][http_acm.hdu.edu.cn_showproblem.php_pid_2141]

[20160726203553997]: /images/20220719/46784d75a4814fc39dee79649b264c8a.png
[20160726203601893]: /images/20220719/89cb5cfda8c94708862fa01d4a10588f.png
[http_acm.hdu.edu.cn_showproblem.php_pid_2141]: http://acm.hdu.edu.cn/showproblem.php?pid=2141