小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,表示一个询问。
接下来T组数据:
接下来第一行一个数N,表示石子个数。(1≤N≤100)
接下来第二行N个数,表示石子的重量。(1≤w_i≤100)
接下来第三行一个数M,表示询问个数。(1≤M≤100)
接下来M行每行一个数k,表示一个询问。
Output
对于每组数据,输出"YES"或者"NO"
Sample Input
1 2 1 4 3 2 4 5
Sample Output
NO YES YES
解题思路: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 long
const 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;
}