砝码称重(dp)

题目链接:https://www.acwing.com/problem/content/3420/

题目
你有一架天平和 $N$ 个砝码，这 $N$ 个砝码重量依次是 $W_1,W_2,\cdot \cdot \cdot ,W_N$。

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

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

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

第二行包含 $N$ 个整数：$W_1,W_2,W_3,\cdot \cdot \cdot ,W_N$。

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

数据范围
对于 $50$ 的评测用例，$1\le N\le 15$。
对于所有评测用例，$1\le N\le 100$，$N$ 个砝码总重不超过 $10^5$。

输入样例：

3
1 4 6

输出样例：

10

样例解释
能称出的 $10$ 种重量是：$1、2、3、4、5、6、7、9、10、11$。

1 = 1；
2 = 6 − 4 (天平一边放 6，另一边放 4)；
3 = 4 − 1；
4 = 4；
5 = 6 − 1；
6 = 6；
7 = 1 + 6；
9 = 4 + 6 − 1；
10 = 4 + 6；
11 = 1 + 4 + 6。

思路:

用 $dp[i][j]$ 表示前 $i$ 个物品能否表示重量 $j$

对于每种物品有 $3$ 种选法，
$1.$ 不选 此时 $dp[i][j]=dp[i-1][j]$
$2.$ 放左边$(+w[i])$ 此时 $dp[i][j]=dp[i-1][j-w[i]]$
$3.$ 放右边$(-w[i])$ 此时 $dp[i][j]=dp[i-1][j+w[i]]$

第一种选法一定有，第二种和第三种要判断是否小于 $-m$ ,是否大于 $m$
因为所有砝码能表示的重量最多为 $-m$ ~ $m$
c++中数组下标不能为负数，而 $-m<=j<=m$ 可以通过添加一个偏移量，使 $0<=j<=2m$

AC代码

#include <iostream>

using namespace std;

const int N = 110, M = 200010, B = M / 2;

int n,m,w[N];
bool dp[N][M];

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        cin >> w[i];
        m += w[i];
    }
    
    dp[0][B] = true;
    for(int i = 1; i <= n; i++)
    {
        for(int j = -m; j <= m; j++)
        {
            dp[i][j + B] = dp[i-1][j + B]; // 不选
            if(j - w[i] >= -m) dp[i][j + B] |= dp[i-1][j - w[i] + B]; // +w[i]
            if(j + w[i] <= m) dp[i][j + B] |= dp[i-1][j + w[i] + B]; // -w[i]
        }
    }
    
    long long ans = 0;
    for(int j = 1; j <= m; j++)
        if(dp[n][j + B]) ans++;
        
    cout << ans << endl;
    
    return 0;
}