蓝桥杯 砝码称重(01背包)

与01背包的区别：每个物品存在正权和负权，都考虑到转移情况即可，同时加一个偏移量防止越界

#include<iostream>
#include<algorithm>
#include<map>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
const int N=105,M=1e5+5;
bool f[N][M*2+5];//f[i][j] 对于前i个砝码，重量为j
int n,m;
int w[N]; 
ll dp()
{	//由于对每个物品存在取负权，加上偏移量防止越界 
	f[0][M]=true;
	for(int i=1;i<=n;i++)
		for(int j=-m;j<=m;j++)//总重量范围为[-m,m] 
		{
			f[i][j+M]=f[i-1][j+M];
			if(j+w[i]<=m)//此处判断条件可以不写 
				f[i][j+M]|=f[i-1][j+w[i]+M];
			if(j-w[i]>=-m)//同上 
				f[i][j+M]|=f[i-1][j-w[i]+M];
		}
	ll res=0;
	for(int i=1;i<=m;i++)
		if(f[n][i+M]) res++;
	return res;
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>w[i],m+=w[i];
	cout<<dp(); 
}