Bootstrap

P2347 [NOIP1996 提高组] 砝码称重(bitset优化dp,bitset用法)--2024冲刺蓝桥杯省一

点击跳转例题
 

思路:将多重背包转化为01背包,因为数据比较小,所以不用二进制优化。
 

01背包解题:
1.定义数组,f(k)表示前 i 个砝码,重量的组合是否可以为 k ,
2.状态转移,看成 01 背包,如果f(i)可以表示,那么f(i+a[i])也可以表示,所以为f(i+a[i])=f(i);
3.初始化:根据定义来初始化f(0)=1。
4.递推,模仿01背包递推:

代码:
 

#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=2e5+10,M=1e3+100,mod=998244353,INF=0x3f3f3f3f;

int w[]={1,2,3,5,10,20};
int f[M];
int a[M];

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

    int n=0;
    int sum=0;
    for(int i=0;i<6;i++)
    {
        int cnt;cin>>cnt;
        for(int j=0;j<cnt;j++)
        {
            a[++n]=w[i];
            sum+=w[i];
        }
    }

    f[0]=1;
    for(int i=1;i<=n;i++)
        for(int j=sum;j>=0;j--)
        {
            f[j+a[i]]=f[j];
        }
    int ans=0;
    for(int i=1;i<=sum;i++)
        if(f[i])ans++;
    cout<<"Total="<<ans<<endl;
    return 0;
}

bitset用法:
 

    //f.set()把所有位置设置为1,f.set(0)把第pos==0位设置为1
    //string s=t.to_string();  转化为字符串;
    //int t=f.test(0); 返回第pos==0位上的值,
    //f.flip();flip()函数反置bitset中所有的位,即将1设为0,0设为1。
    // 如果指定pos,那么只有pos上的位被反置。
    //cout<<f.any()<<endl;  检查bitset上是否存在某一位为1 ;
    // cout<<f.to_ulong()<<endl;  把二进制的f转化为长整形
    //f.count()返回f中1的数量;



bitset,二进制优化:
因为总和比较小,所以我们可以优化为与总和有关,因为bitset的每一位都是用bool表示,所以可以用bitset的每一位是否为1,来表示总和是否可以为bitset的位数,例如:1000表示可以取总和为4。

#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=2e5+10,M=1e3+10,mod=998244353,INF=0x3f3f3f3f;

int w[]={1,2,3,5,10,20};

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

    bitset<1020>f;
    //初始化
    f[0]=1;
    for(int i=0;i<6;i++)
    {
        int cnt;cin>>cnt;

        // | 操作。例子,当w[i]==2时,f为00010时,00010|=01000 ->01010;表示4和2两种重量和都可以取。
        for(int j=0;j<cnt;j++)
            f|=f<<w[i];
    }
    //第0位不算
    cout<<"Total="<<f.count()-1<<endl;

    return 0;
}


 

;