题目:
你有一架天平和 N 个砝码,这 N 个砝码重量依次是 W1, W2, · · · , WN。 请你计算一共可以称出多少种不同的重量? 注意砝码可以放在天平两边。
【输入格式】
输入的第一行包含一个整数 N。
第二行包含 N 个整数:W1, W2, W3, · · · , WN。
【输出格式】
输出一个整数代表答案。
【样例输入】
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。
题目解析:
状态转移方程:a[i][j]=a[i-1][j]||a[i-1][j+w[i]]||a[i-1][abs(j-w[i])];
代码:
#include <bits/stdc++.h>
using namespace std;
int n,w[101],a[101][100002],sum,num;
map<int,int> p;
int main() {
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
cin>>w[i];
num+=w[i];
}
a[0][0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j<=num;j++){
a[i][j]=a[i-1][j]||a[i-1][j+w[i]]||a[i-1][abs(j-w[i])];
}
}
for(int i=1;i<=num;i++){
if(a[n][i]) sum++;
}
cout<<sum;
return 0;
}