清楚姐姐学01背包(Hard Version)

题目如下

题目大意：

对于每一个蝴蝶结，如果不选的总价值价值小于原本的总价值，输出0.否则输出添加多少价值，才会出现上诉情况。

题目思路

首先我们定义

dpf[i][j]，取前i个物品，容积为j的最大总价值。

dpb[i][j],去后i个物品，容积为j的最大总价值。

这两个方程先求出来。

接着定义Vmax为不选第i件物品的最大总价值,Vmaxi为选了第i件物品的最大总价值。

显然Vmax=max(dpf[i-1][k]+dpb[n-i][m-k]);

Vmaxi=max(dp[i-1][k]+dpb[n-i][m-w[i]-k]);

最后我们比较Vmax和Vmaxi即可。

数据说明如果每次都跟01背包一样dp一次，TLE没商量。

代码如下

#include <bits/stdc++.h>
long long dpf[5005][5005];
long long dpb[5005][5005];
long long w[5005], v[5005];

int main() {
    int N, M;
    long long ans[5555];
    scanf("%d%d", &N, &M);
    for (int i = 1; i <= N; i++) {
        scanf("%lld%lld", &w[i], &v[i]);
        for (int j = M; j >= 0; j--) {
            if(j>=w[i])
            dpf[i][j] = std::max(dpf[i - 1][j], dpf[i - 1][j - w[i]] + v[i]);
            else dpf[i][j]=dpf[i-1][j];
        }
    }
    for (int i = 1; i <= N; i++) {
        for (int j = M; j >= 0; j--) {
            if(j>=w[N-i+1])
            dpb[i][j] = std::max(dpb[i - 1][j], dpb[i - 1][j - w[N - i + 1]] + v[N - i + 1]);
            else dpb[i][j]=dpb[i-1][j];
        }
    }
    for (int i = 1; i <= N; i++) {
        long long vmax=0, vmaxi=0;
        
        for(int j=0;j<=M;j++)
        {
            vmax=std::max(dpf[i-1][j]+dpb[N-i][M-j],vmax);    
        }
        for(int j=0;j<=M-w[i];j++)
        {
            vmaxi=std::max(vmaxi,dpf[i-1][j]+dpb[N-i][M-w[i]-j]+v[i]);
        }
        ans[i]=std::max(vmax - vmaxi + 1, (long long)0);
        printf("%lld\n", ans[i]);
    }
}