首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >浙江工业大学校赛 小M和天平

浙江工业大学校赛 小M和天平

作者头像
ShenduCC
发布2018-04-27 11:55:05
5350
发布2018-04-27 11:55:05
举报
文章被收录于专栏:算法修养算法修养

小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;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017-03-22 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 小M和天平
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档