ET算法【背包】

一、小段话

今天是2024.2.17，要回学校了，过了一个寒假，我好像还是一样的蒟蒻。开始学dp了，但是好hard，学不会就只能坐牢。

二、01背包

P1026 - 采药 - ETOJ (eriktse.com)

1.基础版

dp[i][j]表示选到第i个物品已经用了j时间的价值。

对于每一个物品分为选和不选两种状态，选：dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - t] + v)，不选：dp[i][j] = dp[i - 1][j]

初始状态为第1个物品，价值均为0，终止状态为选到第M个物品

第二层j的循环可以理解为给你j的容量（j从0一直到无穷），在当前选到第i个物品时，最大的价值会是多少

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e3 + 10;
ll dp[N][N], t[N], v[N];

int main() {
	int T, M; 
	while(cin >> T >> M && (T != 0 && M != 0)){
		for(int i = 1; i <= M; i++) cin >> t[i] >> v[i];		
		for(int i = 1; i <= M; i++){
			for(int j = 0; j <= T; j++){
				if(j >= t[i]) dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - t[i]] + v[i]);
				else dp[i][j] = dp[i - 1][j];
			}
		}
		cout << dp[M][T] << '\n';
	}
}

2.优化版

其实只用了此时的状态和上一个状态的数据，所以可以进行优化，变成dp[2][N]

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e3 + 10;
ll dp[2][N], t[N], v[N];

int main() {
	int T, M; 
	while(cin >> T >> M && (T != 0 && M != 0)){
		for(int i = 1; i <= M; i++) cin >> t[i] >> v[i];
		memset(dp, 0, sizeof dp);	
		for(int i = 1; i <= M; i++){
			int y = i & 1; //y表示当前行
			for(int j = 0; j <= T; j++){
				if(j >= t[i]) dp[y][j] = max(dp[y ^ 1][j], dp[y ^ 1][j - t[i]] + v[i]);
				else dp[y][j] = dp[y ^ 1][j];
			}
		}
		cout << dp[M & 1][T] << '\n';
	}
}

3.终极版

滚动数组优化中用到的是两个数组，如果先把上一个状态拷贝下来，再进行状态转移，也是可以的，所以不拷贝，直接在原数组上接着操作即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e3 + 10;
ll dp[N], t[N], v[N];

int main() {
	int T, M; 
	while(cin >> T >> M && (T != 0 && M != 0)){
		for(int i = 1; i <= M; i++) cin >> t[i] >> v[i];
		memset(dp, 0, sizeof dp);	
		for(int i = 1; i <= M; i++){
			for(int j = T; j >= t[i]; j--){ 
                //从后往前，前面的是上一个状态，后面的是新的状态
				dp[j] = max(dp[j], dp[j - t[i]] + v[i]);
			}
		}
		cout << dp[T] << '\n';
	}
}

三、完全背包

P1024 - 无穷背包 - ETOJ (eriktse.com)

可以选无穷多个的物品

1.基础版

dp[i][j]表示到第i个物品为止，用了j的体积，获得物品的最大价值

对于dp[i][j]的取值，先由上一个状态拷贝，即dp[i][j] = dp[i - 1][j]（就是不选此物品），也可以由本状态转移过来，即dp[i][j] = dp[i][j - v[i]] + w[i]（因为可以重复选，所以是本状态i）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
ll dp[510][N], w[510], v[510];

int main() {
	int m, n; cin >> m >> n;
	for(int i = 1; i <= n; i++) cin >> w[i] >> v[i];
	for(int i = 1; i <= n; i++){
		for(int j = 0; j <= m; j++){
			if(j >= v[i])dp[i][j] = max(dp[i - 1][j], dp[i][j - v[i]] + w[i]);
			else dp[i][j] = dp[i - 1][j];
		}
	}
	cout << dp[n][m] << '\n';
}

但是交一发却过不了，这是因为数组空间爆了，需要压缩优化

2.优化版

因为本状态要么是从上一个状态转移过来，要么是从本状态的左侧转移过来，所以可以滚动数组优化

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
ll dp[2][N], w[510], v[510];

int main() {
	int m, n; cin >> m >> n;
	for(int i = 1; i <= n; i++) cin >> w[i] >> v[i];
	for(int i = 1; i <= n; i++){
		int y = i & 1;
		for(int j = 0; j <= m; j++){
			if(j >= v[i])dp[y][j] = max(dp[y ^ 1][j], dp[y][j - v[i]] + w[i]);
			else dp[y][j] = dp[y ^ 1][j];
		}
	}
	cout << dp[n & 1][m] << '\n';
}

3.终极版

既然是从本状态转移过来，那么可以再优化一下

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
ll dp[N], w[510], v[510];

int main() {
	int m, n; cin >> m >> n;
	for(int i = 1; i <= n; i++) cin >> w[i] >> v[i];
	for(int i = 1; i <= n; i++){
		for(int j = v[i]; j <= m; j++){
			dp[j] = max(dp[j], dp[j - v[i]] + w[i]);
		}
	}
	cout << dp[m] << '\n';
}

四、多重背包

P1103 - 多重背包 - ETOJ (eriktse.com)

1.转化为01背包

将多个的物品转化为相同的同种物品，再跑01背包即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e2 + 10;
ll dp[N];

int main() {
	ll m, n; cin >> m >> n;
	for(int i = 1; i <= n; i++){
		int s, w, v; cin >> s >> w >> v;
		while(s--){
			for(int j = m; j >= v; j--){
				dp[j] = max(dp[j], dp[j - v] + w);
			}
		}
	}
	cout << dp[m];
}

2.多重背包的二进制优化

P1104 - 多重背包二周目 - ETOJ (eriktse.com)

如果数据量够大，第一种方法就爆时间复杂度了，问题主要是分解成的单个物品太多了，所以优化为用二进制来表示所有可能取到的个数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 2e3 + 10;
ll dp[N];

int main() {
	ll m, n; cin >> m >> n;
	for(int i = 1; i <= n; i++){
		int s, w, v; cin >> s >> w >> v;
		ll x = 1;
		vector<ll> vec;
		while(s >= x) vec.push_back(x), s -= x, x <<= 1;
		if(s) vec.push_back(s);
		for(auto &k : vec){
			for(int j = m; j >= k * v; j--){
				dp[j] = max(dp[j], dp[j - k * v] + k * w);
			}	
		}			
	}
	cout << dp[m];
}

3.多重背包的单调队列优化

to be continue ->

五、例题

1.01背包（div3 rating *1800）

Problem - E - Codeforces

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;
typedef double db;
#define int ll
const int N = 2e5 + 10;
int c[N],h[N];

void solve() {
	int m, x; cin >> m >> x;
	int sum = 0;
	vector<int> dp(1);
	for(int i = 1; i <= m; i++){
		cin >> c[i] >> h[i];
		sum += h[i];
	}
	dp.resize(sum + 1, -1);
	for(int i = 1; i <= m; i++){
		for(int j = sum; j >= h[i]; j--){
			if(dp[j - h[i]] >= c[i]){
				dp[j] = max(dp[j], dp[j - h[i]] - c[i]);
			}
		}
		for(int j = 0; j <= sum; j++){
			if(dp[j] >= 0) dp[j] += x;
		}
	}
	int ans = 0;
	for(int i = 0; i < dp.size(); i++){
		if(dp[i] >= 0){
			ans = i;
		}
	}
	cout << ans << '\n';
}

signed main() {
	qio
	int T = 1;
	cin >> T;
	while (T--)	solve();
}