题目如下
题目大意:
对于每一个蝴蝶结,如果不选的总价值价值小于原本的总价值,输出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]);
}
}