P8742 [蓝桥杯 2021 省 AB] 砝码称重--2024蓝桥杯冲刺省一

点击跳转例题

建议使用bitset优化来做，因为比较简介，而且易懂，当然也可以用01背包来做。简单版参考：P2347 [NOIP1996 提高组] 砝码称重（bitset优化dp，bitset用法）

bitset：
 

#include <bits/stdc++.h>
#define int long long //(有超时风险)
#define PII pair<int,int>
#define endl '\n'
#define LL __int128

using namespace std;

const int N=2e6+10,M=1e3+10,mod=998244353,INF=0x3f3f3f3f;

int a[N],b[N],c[N],pre[N];
//dp[j]|=dp[j-a[i]],dp[j]|=dp[j+a[i]]

//然后不难发现这个转移在位运算中下标减相当于右移，下标加相当于左移，
//所以涉及的所有操作都是位运算，于是直接上 bitset 优化 DP，
//将转移过程的加减改成位运算即可。
//比如将第0为设置为1，那么0+4==4,就是f>>4相当于第4位变成了1，那么就表示能取到4
//最后减去第0个位置上的1就行了，因为0不计入答案；
signed main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    bitset<100010>f;

    int n;cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    
    f[0]=1;//f.set(0);
    for(int i=1;i<=n;i++)
        f|=f<<a[i];
    for(int i=1;i<=n;i++)
        f|=f>>a[i];

    cout<<f.count()-1<<endl;

    return 0;
}

01背包，f（i，j）表示前 i 个物品总和为 j 能否表示

 

 #include <bits/stdc++.h>
 #define int long long //(有超时风险)
 #define PII pair<int,int>
 #define endl '\n'

 using namespace std;

 //数组太大会超内存
 const int N=1e5+5e4,M=1e2+10,mod=998244353,INF=0x3f3f3f3f;

 int w[M];
 int f[M][N];

 signed main()
 {
 	std::ios::sync_with_stdio(false);
     std::cin.tie(nullptr);

 	int n,m=0;cin>>n;

 	for(int i=1;i<=n;i++)
 	{
 		cin>>w[i];
 		m+=w[i];
 	}

 	//f（i，j）表示前i个物品是否可以选到和为j
 	for(int i=1;i<=n;i++)
 	{
 		for(int j=1;j<=m;j++)
 		{
 			f[i][j]=f[i-1][j];
 			if(j==w[i])f[i][j]=1;
 			else if(f[i-1][abs(j-w[i])])f[i][j]=1;
 			else if(f[i-1][j+w[i]])f[i][j]=1;
 		}
 	}
 	int ans=0;
 	for(int i=1;i<=m;i++)
 	{
 		if(f[n][i])ans++;
 	}
 	cout<<ans<<endl;
 	return 0;
  }