长沙理工大学第十二届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 100000int 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");        }    }}