背包问题
-
首先这个题目涉及到了背包问题,不理解的同学们需要去先学习一下。
-
这是个比较典型的 0 − 1 0-1 0−1 背包问题。只不过多了一维我们当前武器的状态。设定 f [ i ] [ j ] [ k ] f[i][j][k] f[i][j][k] 表示考虑前 i i i 个怪物,体积小于等于 j j j 并且当前的武器是 k k k 的最大价值。那么这个题目就是在原来背包转移的基础上,多了一个武器的转移,因为武器是只能按照题干的顺序变那么就有以下转移方程:
当不选择第 i i i 个物品的时候:
f [ i ] [ j ] [ k ] = m a x ( f [ i − 1 ] [ j ] [ k ] , f [ i − 1 ] [ j ] [ k − 1 ] ) ; f[i][j][k] = max(f[i - 1][j][k], f[i - 1][j][k - 1]); f[i][j][k]=max(f[i−1][j][k],f[i−1][j][k−1]);
当选择第 i i i 个物品的时候:
f [ i ] [ j ] [ k ] = m a x ( f [ i − 1 ] [ j − v [ i ] ] [ k ] , f [ i − 1 ] [ j − v [ i ] ] [ k − 1 ] ) + w [ i ] ; f[i][j][k] = max(f[i - 1][j - v[i]][k], f[i - 1][j - v[i]][k - 1]) + w[i]; f[i][j][k]=max(f[i−1][j−v[i]][k],f[i−1][j−v[i]][k−1])+w[i];
其中 v [ i ] v[i] v[i] 表示的是杀死第 i i i 个怪物的所消耗的体力值。 w [ i ] w[i] w[i] 表示杀死第 i i i 个怪物的得分。
- 因为是 0 − 1 0-1 0−1 背包,当前可以优化掉一维。下面的代码是没有优化的。
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
const int N = 3010;
int v[N], w[N][5], f[N][N][5];
void solve() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i ++ ) {
cin >> v[i];
}
for (int i = 1; i <= n; i ++ ) {
for (int j = 1; j <= 4; j ++ ) {
cin >> w[i][j];
}
}
for (int i = 1; i <= n; i ++ ) {
for (int j = 1; j <= m; j ++ ) {
for (int k = 1; k <= 4; k ++ ) {
f[i][j][k] = max(f[i - 1][j][k], f[i - 1][j][k - 1]);
if (j >= v[i]) {
f[i][j][k] = max(f[i - 1][j - v[i]][k] + w[i][k], f[i][j][k]);
f[i][j][k] = max(f[i - 1][j - v[i]][k - 1] + w[i][k], f[i][j][k]);
}
}
}
}
cout << max({f[n][m][1], f[n][m][2], f[n][m][3], f[n][m][4]}) << endl;
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int T = 1;
// cin >> T;
while (T--) {
solve();
}
return 0;
}