与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();
}