小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动态规划#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 "); continue; } if(k==0) { printf("YES "); continue; } if(ans[k]==1) printf("YES "); else printf("NO "); } } return 0; }