牛客 -- 小M和天平(简单dp)

小M和天平

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

问：给你n个砝码，你是否能称出来重量为k的物品。

问题类似于21年蓝桥杯的 “砝码称重” 这个题，一个问重量为k能不能称出来，一个问能称出来多少个不同的重量k

设dp[i][j] ,表示前i个砝码能否称出重量为j的物品( 0 or 1 )，注意初始化
类似于背包，第i个物品，选或不选。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int dp[110][11000],a[110];
int main()
{
    int n,sum;
    while(scanf("%d",&n)!=EOF)
    {
       sum=0;
        memset(dp,0,sizeof(dp));
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        for(int i=0; i<=n; i++)
            dp[i][0]=1;     //初始化   
        
        // dp[i][j]=1  表示 前i个砝码能够称出重量 j 的物品
        for(int i=1; i<=n; i++)
        {
            for(int j=10000; j>=0; j--) //j要到0,j=0对应的情况为单独放第i个砝码.
            {
                if(dp[i-1][j]) 
                {
                    dp[i][j+a[i]]=1;
                    dp[i][abs(j-a[i])]=1;  // 使用第i个砝码，分别放在天平的两端
                    dp[i][j]=1;   // 不使用第i个砝码,前i-1个砝码能实现的重量，前i个必然行 
                }
            }
        }
         int t,m;
        scanf("%d",&t);
        while(t--)
        {
            scanf("%d",&m);
            if(m<=sum&&dp[n][m])
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}