小M和天平

                                                                             Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
                                                                                                           Total Submission(s): 568    Accepted Submission(s): 108

Problem Description

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

 

Input

第一行T(1≤T≤100)，表示T组数据。
接下来T组数据：
接下来第一行一个数N，表示石子个数。（1≤N≤100）
接下来第二行N个数，表示石子的重量。（1≤w_i≤100）
接下来第三行一个数M，表示询问个数。（1≤M≤100）
接下来M行每行一个数k，表示一个询问。

 

Output

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

 

Sample Input

121  43245

 

Sample Output

NOYESYES

 

解题思路：dp（注意因为砝码左右都可以放，所以不能将这些砝码全放一边来判断能否称出来），dp[i][j]表示第i件物品能否表示出j的重量

#include <iostream>#include <cstdio>#include <string>#include <cstring>#include <algorithm>#include <cmath>#include <queue>#include <vector>#include <set>#include <bitset>#include <stack>#include <map>#include <climits>#include <functional>using namespace std;#define LL long longconst int INF=0x3f3f3f3f;int t,m,n,w[110],dp[110][20010],x;int main(){    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        memset(dp[0],0,sizeof dp[0]);        dp[0][10000]=1;        for(int i=1;i<=n;i++)        {            scanf("%d",&w[i]);            for(int j=0;j<20010;j++) dp[i][j]=dp[i-1][j];            for(int j=0;j<20010-w[i];j++0) dp[i][j+w[i]]|=dp[i-1][j];            for(int j=w[i];j<20010;j++) dp[i][j-w[i]]|=dp[i-1][j];        }        scanf("%d",&m);        while(m--)        {            scanf("%d",&x);            if (x>1e4) puts("NO");            else puts(dp[n][R+x]||dp[n][R-x]?"YES":"NO");        }    }    return 0;}