长沙理工大学第十二届ACM大赛 D 小M和天平

题目描述

小M想知道某件物品的重量，但是摆在他面前的只有一个天平（没有游标）和一堆石子，石子可以放左边也可以放右边。他现在知道每个石子的重量。问能不能根据上述条件，能不能测出所问的重量。

输入描述:

多组数据，第一行一个数N，表示石子个数。（1<=N<=100） 接下来第二行N个数，表示石子的重量。（1<=Wi<=100） 接下来第三行一个数M，表示询问个数。（1<=M<=1000） 接下来M行每行一个数k(1<=k<=1e9)，表示一个询问。

输出描述:

对于每组数据，输出"YES"或者"NO"

示例1

题目本身挺难的。。

可是数据给的太小了。。

通过分析数据可以得出石子的总质量最大值不会查过1w

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100000
int vis[N+5],b[N+5],x,k;
int main()
{
    int n,m;
    while(~scanf("%d",&n))
    {
        memset(vis,0,sizeof(vis));
        k = 0;
        for(int i = 0; i < n; i++)
        {
            scanf("%d",&x);
            int l = k;
            for(int j = 0; j < l; j++)
            {
                if(vis[x+b[j]] == 0)
                {
                    vis[x+b[j]] = 1;
                    b[k++] = x+b[j];
                }
                if( b[j] - x > 0 && vis[b[j]-x] == 0)
                {
                    vis[b[j]-x] = 1;
                    b[k++] = b[j] - x;
                }
                if( x-b[j] > 0 && vis[x-b[j]] == 0)
                {
                    vis[x-b[j]] = 1;
                    b[k++] = x-b[j];
                }
            }
            if(vis[x] == 0)
            {
                vis[x] = 1;
                b[k++] = x;
            }
        }
        /*for(int i = 0; i < k; i++)
            printf("b[%d] = %d\n",i,b[i]);*/
        scanf("%d",&m);
        for(int i = 0; i < m; i++)
        {
            scanf("%d",&x);
            if( x > 100000 || !vis[x])
                printf("NO\n");
            else
                printf("YES\n");
        }
    }
}