题目链接:https://www.acwing.com/problem/content/3420/
题目
你有一架天平和
N
N
N 个砝码,这
N
N
N 个砝码重量依次是
W
1
,
W
2
,
⋅
⋅
⋅
,
W
N
W_1,W_2,⋅⋅⋅,W_N
W1,W2,⋅⋅⋅,WN。
请你计算一共可以称出多少种不同的正整数重量?
注意砝码可以放在天平两边。
输入格式
输入的第一行包含一个整数
N
N
N。
第二行包含 N N N 个整数: W 1 , W 2 , W 3 , ⋅ ⋅ ⋅ , W N W_1,W_2,W_3,⋅⋅⋅,W_N W1,W2,W3,⋅⋅⋅,WN。
输出格式
输出一个整数代表答案。
数据范围
对于
50
50%
50 的评测用例,
1
≤
N
≤
15
1≤N≤15
1≤N≤15。
对于所有评测用例,
1
≤
N
≤
100
1≤N≤100
1≤N≤100,
N
N
N 个砝码总重不超过
1
0
5
10^5
105。
输入样例:
3
1 4 6
输出样例:
10
样例解释
能称出的
10
10
10 种重量是:
1
、
2
、
3
、
4
、
5
、
6
、
7
、
9
、
10
、
11
1、2、3、4、5、6、7、9、10、11
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。
思路:用 d p [ i ] [ j ] dp[i][j] dp[i][j] 表示前 i i i 个物品能否表示重量 j j j
对于每种物品有
3
3
3 种选法,
1.
1.
1. 不选 此时
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
]
dp[i][j] = dp[i - 1][j]
dp[i][j]=dp[i−1][j]
2.
2.
2. 放左边
(
+
w
[
i
]
)
(+w[i])
(+w[i]) 此时
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
−
w
[
i
]
]
dp[i][j] = dp[i - 1][j - w[i]]
dp[i][j]=dp[i−1][j−w[i]]
3.
3.
3. 放右边
(
−
w
[
i
]
)
(-w[i])
(−w[i]) 此时
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
+
w
[
i
]
]
dp[i][j] = dp[i - 1][j + w[i]]
dp[i][j]=dp[i−1][j+w[i]]
第一种选法一定有,第二种和第三种要判断是否小于
−
m
-m
−m ,是否大于
m
m
m
因为所有砝码能表示的重量最多为
−
m
-m
−m ~
m
m
m
c++中数组下标不能为负数,而
−
m
<
=
j
<
=
m
-m <= j <= m
−m<=j<=m 可以通过添加一个偏移量,使
0
<
=
j
<
=
2
m
0 <= j <= 2m
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;
}