牛客竞赛——小M和天平(简单dp)

题目链接:https://ac.nowcoder.com/acm/problem/13586

 思路:首先所有能称出的重量均在0~10000之间

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <string>
#include <stack>
#include <set>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
typedef long long ll;

namespace ak{
    //dp[i][j]前i个石子能否称出j的重量
    int dp[110][10010];
    int n,m,k;
    int a[110];
    //核心dp
    void getdp(){
        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        for(int i=1;i<=n;i++){
            //所有可能称出的重量不可能超过10000;
            for(int j=0;j<=10000;j++){
                if(dp[i-1][j]){//如果前i-1个石子能称出j的重量
                    dp[i][j]=1;//不放该石子
                    dp[i][j+a[i]]=1;//放叠加的一边
                    dp[i][abs(j-a[i])]=1;//放抵消的一边
                }
            }
        }
    }
    void query(){
        cin>>m;
        while(m--){
            cin>>k;
            if(k<=10000&&dp[n][k]){
                cout<<"YES"<<endl;
            }
            else cout<<"NO"<<endl;
        }
    }
    void work(){
        while(cin>>n){
            for(int i=1;i<=n;i++){
              cin>>a[i];
            }
            getdp();
            query();
        }
    }
}

int main(){
    IOS;
    ak::work();
    getchar();
    getchar();
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值