砝码称重
你有一架天平和 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;
}