DP(背包问题) - 砝码称重 - 第十二届蓝桥杯省赛第一场C++A/B组

DP(背包问题) - 砝码称重 - 第十二届蓝桥杯省赛第一场C++A/B组

题意：

你有一架天平和 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 ≤ N ≤ 15。
对于所有评测用例，1 ≤ N ≤ 100，N 个砝码总重不超过 10⁵。

输入样例：

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。

---

分析：

有限制的选择问题，即背包问题。

每种物品有3种状态：不选、放左边（权值为正）、放右边（权值为负）。

状态转移方程：

$f[i][j]：考虑前i件物品，是否能够称出的质量为j的物品。记：$

$$m=\sum _{i=1}^n{W}_i$$

$①、不选第i件物品，则f[i][j]=f[i-1][j]。$

$②、第i件物品放左边，则f[i][j]=f[i-1][j-w[i]]。$

$③、第i件物品放右边，则f[i][j]=f[i-1][j+w[i]]。$

$其中j\in [-m,m]，为了防止数组越界，我们给j加上偏移量m，即j\in [0,2m]。$

$最后统计f[n][j]（j\in [1,2m]）中，能够称出的不同质量的个数。$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;

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

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

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>w[i], m += w[i];
    
    f[0][B] = true;
    for(int i=1;i<=n;i++)
        for(int j=-m;j<=m;j++)
        {
            f[i][j+B] = f[i-1][j+B];
            if(j - w[i] >= -m) f[i][j+B] |= f[i-1][j-w[i]+B];
            if(j + w[i] <= m) f[i][j+B] |= f[i-1][j+w[i]+B];
        }
    
    int res = 0;
    for(int i = B+1; i <= m + B; i ++)
        if(f[n][i])
            res ++;
    
    cout<<res<<endl;
    
    return 0;
}