01背包模板

一、01背包的概念

有n种物品，每种物品有它对应的价值和重量，第i种物品重量为wi，价值为vi，每种物品只有一件，你有容量为m的背包，怎么选取在不超重的情况下取得最大价值。

我们使用 dp[i][j] 表示第i件物品，容量为j时的最大价值，每件物品有取和不取的状态，当前最大价值取决于取该物品，和不取该物品的最大价值（个人理解），得到下面的方程

二、例题及其优化

洛谷：采药
链接:https://www.luogu.com.cn/problem/P1048

//普通二维数组，时间复杂度和空间复杂度都达到了o(nm)
#include <iostream>
using namespace std;
const int mx = 1e3 + 6;
int n, m, dp[mx][mx], w[mx], v[mx];
void solve() {
	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 < w[i])
				dp[i][j] = dp[i - 1][j];
			else {
				dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - w[i]] + v[i]);
			}
		}
	}
	cout << dp[n][m];
}
int main()
{
	solve();
	return 0;
}

普通二维数组dp可以过，这题数据较弱。接下来采用一维数组优化

#include <iostream>
#include <cstring>
using namespace std;
void solve() {
	int t, m, f[1005], w[1005], v[1005];
	memset(f, 0, sizeof f);
	cin >> t >> m;
	for (int i = 1; i <= m; i++)  cin >> w[i] >> v[i];
	for (int i = 1; i <= m; i++) {
		for (int j = t; j >= w[i]; j--) {
			f[j] = max(f[j], f[j - w[i]] + v[i]);
		}
	}
	cout << f[t];
}
int main()
{
	solve();
	return 0;
}

三、模板

//n是种数 m是容量 该问题为01的不必完全装满，数组全初始化为0
//最好开long long，有些题可能会爆
scanf("%d %d", &n,&m);
	for (int i = 1; i <= n; i++)
		scanf("%d %d", &v[i], &w[i]);
	for (int i = 1; i <= n; i++)
		for (int j = m; j >= w[i]; j--)
			f[j] = max(f[j], f[j - w[i]] + v[i]);
	printf("%d", f[m]);

如果题目要求刚好装满

//初始化有所不同
#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
const int mx = 1e4 + 9;
int n, m;
int f[mx], w[mx], v[mx];
void solve() {
    cin >> m >> n;
    for(int i = 0; i < mx; i++) f[i] = -inf;
    f[0] = 0;
    for(int i = 1; i <= n; i++) {
        cin >> v[i] >> w[i];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = m; j >= v[i]; j--) {
            f[j] = max(f[j], f[j - v[i]] + w[i]);
            if (f[j] < 0)
                f[j] = -inf;
        }
    }
    if(f[m] > 0) cout << f[m];
    else       cout << "no";
}
int main() {
    solve();
    return 0;
}