小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
1
2
1 4
3
2
4
5
 
Sample Output
NO
YES
YES
动态规划
#include <iostream>
#include <string.h>
#include <math.h>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h> using namespace std;
typedef long long int LL;
const int maxn=1e5;
int w[105];
int n;
int m;
int dp[10005][105];
int ans[10005];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
int sum=0;
int cnt=1;
memset(dp,0,sizeof(dp));
memset(ans,0,sizeof(ans));
for(int i=0;i<=100;i++)
dp[0][i]=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);
sum+=w[i];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=sum;j++)
{
if(dp[j][i-1]==1)
dp[j][i]=1;
if(w[i]-j>=0&&dp[w[i]-j][i-1]==1)
dp[j][i]=1;
if(j-w[i]>=0&&dp[j-w[i]][i-1]==1)
{
dp[j][i]=1;
if(j-2*w[i]>0)
dp[j-2*w[i]][i]=1;
} }
}
for(int i=1;i<=sum;i++)
{
for(int j=1;j<=n;j++)
{ if(dp[i][j]==1)
{
ans[i]=1; break;
}
} }
scanf("%d",&m);
int k;
for(int i=1;i<=m;i++)
{
scanf("%d",&k);
if(k>sum)
{
printf("NO\n");
continue;
}
if(k==0)
{
printf("YES\n");
continue;
}
if(ans[k]==1)
printf("YES\n");
else
printf("NO\n");
} }
return 0;
}

 
04-28 18:22