hdu 2141 Can you find it?【二分】

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2141


思路:这道题数据范围小,可以暴力,用map存会超内存,用三层for循环会TLE,所以我们可以先让两组物品的费用相加(成为和费用),然后从小到大排序,枚举每个费用再二分和费用找到相等的;

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<string>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#include<math.h>
using namespace std;
typedef long long ll;
const double pi=acos(-1.0);
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
const int N=502;

int a[N],b[N],c[N];
int sum[N*N];

int found(int x,int l,int r,int q)//二分
{
    while(r>l)
    {
        int mid=(r+l)/2;
        if(x+sum[mid]>=q)
        {
            if(x+sum[mid]==q) return 1;//找到直接return 1
            r=mid;
        }
        else
        {
            l=mid+1;
        }
    }
    return 0;
}

int main()
{
    int l,n,m,cas=0;
    while(~scanf("%d %d %d",&l,&n,&m))
    {
        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]);
        int k,q;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
                sum[i*m+j]=b[i]+c[j];
        }
        sort(sum,sum+n*m);
        scanf("%d",&k);
        printf("Case %d:\n",++cas);
        for(int i=0; i<k; i++)
        {
            int flag=0;
            scanf("%d",&q);
            for(int i=0;i<l;i++)
            {
                if(found(a[i],0,n*m-1,q))
                {
                    flag=1;
                    break;
                }
            }
            if(flag) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41984014/article/details/88916938