蓝桥杯2021年砝码称重

蓝桥杯砝码称重
【问题描述】
你有一架天平和N 个砝码，这N 个砝码重量依次是W1, W2…WN。
请你计算一共可以称出多少种不同的重量？
注意砝码可以放在天平两边。

数据范围
对于 50% 的评测用例，1≤N≤15。
对于所有评测用例，1≤N≤100，N 个砝码总重不超过 1e5。
都是整数

样例：

input：
3
1 4 6
output：
10

01背包问题的小变形，先考虑一边砝码正的重量，再考虑一边重量的负贡献，仅用一维的数组优化空间和代码，AC代码如下：

//砝码称重
#include <iostream>
using namespace std;
#define _for(i,a,b) for(int i=(a);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
const int maxn = 100+5;
const int maxw = 100000+5;
const int inf = 0x3f3f3f3f;
int n,sum,ans;
int w[maxn]; 
bool dp[maxw];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>w[i];
		sum += w[i];
	}
	dp[0] = true;
	rep(i,1,n){
		for(int j=sum;j>=w[i];j--){
			dp[j] |= dp[j-w[i]];
			
		}
	}
	rep(i,1,n){
		rep(j,1,sum-w[i]){
			dp[j] |= dp[j+w[i]];
			
		}
	}
	//
//	rep(i,1,sum){
//		cout<<dp[i]<<" ";
//	} 
//	cout<<endl;
	
	rep(i,1,sum){
		if(dp[i]){
			ans++;
		}
	} 
    cout<<ans<<endl;
	return 0;
}