练习21，砝码称重【动态规划DP】

砝码称重

3417. 砝码称重 - AcWing题库

你有一架天平和 N 个砝码，这 N 个砝码重量依次是 W1,W2,⋅⋅⋅,WN。

请你计算一共可以称出多少种不同的正整数重量？

注意砝码可以放在天平两边。

输入格式
输入的第一行包含一个整数 N。

第二行包含 N 个整数：W1,W2,W3,⋅⋅⋅,WN。

输出格式
输出一个整数代表答案。

数据范围
对于 50% 的评测用例，1≤N≤15。
对于所有评测用例，1≤N≤100，N 个砝码总重不超过 1e5

题解

d[i][j]表示前i个砝码，称出的重量为j

因为在左右两边称出相同的重量是一种方法，所以d[i][j]=d[i][-j]

故而d[i][j-w[i]]=d[i][w[i]-j]=d[i][abs(j-w[i])],避免数组下标为负数的情况

#include<bits/stdc++.h>
using namespace std;

int w[110];
bool d[110][200010];
int main(){
    int n,sum=0,ans=0;
    cin >> n;
    for(int i=1;i<=n;i++){
        cin >> w[i];
        sum+=w[i];
    }
    d[0][0]=1;
    for(int i=1;i<=n;i++){
        for(int j=0;j<=sum;j++)
            d[i][j]=(d[i-1][j] || d[i-1][j+w[i]] || d[i-1][abs(j-w[i])]);
            //只要这三个之一能称出，d[i][j]就能称出
    }
    for(int i=1;i<=sum;i++)
        if(d[n][i]==1) ans++;
    cout << ans;
    return 0;
}