2023牛客寒假算法基础集训营-清楚姐姐学01背包

题目

 思路

用数组dp2[k][i][j]表示考虑第i个蝴蝶结、背包容量为j且不装入第k个蝴蝶结的最大值。假设w[i]表示第i个蝴蝶结的体积，v[i]表示第i个蝴蝶结的价值。其状态转移方程为：首先dp2[k][i][j]=dp2[k][i-1][j]，然后dp2[k][i][j]=max(dp2[k][i][j],dp2[k][i-1][j-w[i]]+v[i])【j>=w[i]】，因为不装入第k个蝴蝶结，所以当i==k时dp2[k][i][j]只能等于dp2[k][i-1][j]。然后我们再用dp[i][j]做一次普通的01背包，然后用flag[i][j][k]表示考虑第i个蝴蝶结，背包容量为j时第k个蝴蝶结有没有被使用过，状态转移为：dp[i][j]<dp[i-1][j-w[i]]+v[i]时，flag[i][j][k]=flag[i-1][j-w[i]][k]且flag[i][j][i]=1，两者相等时，说明两者交集才是必须蝴蝶结。最后如果flag[n][m][i]==1，说明其本来就是必须蝴蝶结，输出0，否则，输出dp[n][m]-dp2[i][n][m-w[i]]-v[i]+1。

代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
int dp[105][105];
int w[105],v[105];
bool flag[105][105][105];
int dp2[105][105][105];
signed main(){
    int n,m;cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>w[i]>>v[i];
    }
    for(int k=1;k<=n;k++){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                dp2[k][i][j]=dp2[k][i-1][j];
                if(i==k)continue;
                if(j>=w[i])dp2[k][i][j]=max(dp2[k][i][j],dp2[k][i-1][j-w[i]]+v[i]);
            }
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            dp[i][j]=dp[i-1][j];
            if(j>=w[i]){
                if(dp[i][j]<dp[i-1][j-w[i]]+v[i]){
                    for(int k=1;k<=n;k++)flag[i][j][k]=flag[i-1][j-w[i]][k];
                    flag[i][j][i]=1;
                    dp[i][j]=max(dp[i][j],dp[i-1][j-w[i]]+v[i]);
                    continue;
                }
                else if(dp[i][j]==dp[i-1][j-w[i]]+v[i]){
                   for(int k=1;k<=n;k++)
                       if(flag[i-1][j][k]&&flag[i-1][j-w[i]][k])flag[i][j][k]=1;
                    continue;
                }
            }
            for(int k=1;k<=n;k++)flag[i][j][k]=flag[i-1][j][k];
        }
    }
    int ans=dp[n][m]+1;
    for(int i=1;i<=n;i++){
        if(flag[n][m][i])cout<<0<<endl;
        else cout<<ans-dp2[i][n][m-w[i]]-v[i]<<endl;
    }
}