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

点击跳转例题

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

bitset:
 

#include 
#define int long long //(有超时风险)
#define PII pair
#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];

    cout<

01背包,f(i,j)表示前 i 个物品总和为 j 能否表示

 

 #include 
 #define int long long //(有超时风险)
 #define PII pair
 #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<

你可能感兴趣的:(动态规划,算法,c++,蓝桥杯)