I. Steadily Growing Steam
思路:滚动数组优化的背包问题,维护一下差值,把数组的负数平移到正数来
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
typedef pair<int, int> PII;
const int N = 110;
ll f[2][N][6010]; // 第 i 个物品 施法 j 次 差值为 k 得到的最大值
ll v[N];
int t[N];
int n, k;
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
#ifndef ONLINE_JUDGE
freopen("D:/Cpp/program/Test.in", "r", stdin);
freopen("D:/Cpp/program/Test.out", "w", stdout);
#endif
cin >> n >> k;
for(int i = 1; i <= n ; i ++) {
cin >> v[i] >> t[i];
}
//一开始什么都不选 f[0][0][0] = 0; 其余都是不可达状态赋值为负无穷
for(int i = 0; i <= k; i ++)
for(int j = 0; j <= 2600; j ++)
f[0][i][j] = -1e18;
f[0][0][0] = 0;
for(int i = 1; i <= n; i ++)
for(int j = 0; j <= k; j ++)
for(int k = 0; k <= 2600; k ++) {
int x = i & 1; //滚动数组
f[x][j][k] = f[x ^ 1][j][k]; //这是一定需要的 不能写成MAX,因为只能从上一层转移来
if(k + t[i] <= 2600)
f[x][j][k] = max(f[x][j][k], f[x ^ 1][j][k + t[i]] + v[i]);
if(k + 2 * t[i] <= 2600 && j)
f[x][j][k] = max(f[x][j][k], f[x ^ 1][j - 1][k + 2 * t[i]] + v[i]);
if(k >= t[i])
f[x][j][k] = max(f[x][j][k], f[x ^ 1][j][k - t[i]] + v[i]);
if(k >= 2 * t[i] && j)
f[x][j][k] = max(f[x][j][k], f[x ^ 1][j - 1][k - 2 * t[i]] + v[i]);
}
// for(int i = 1; i <= 2600; i ++) cout << f[n & 1][k][i] << '\n';
cout << f[n & 1][k][1300] << '\n';
}