背包是 DP 中一类重要而特殊的模型。本文将引导你学习各类背包。
关于背包问题有一个特别著名的讲义:崔添翼的背包九讲。想要学好更多复杂的动态规划问题,对背包问题的深入理解必不可少。你可以选择将背包九讲和本文搭配着阅读。
01 背包
这类问题的形式是:
- 你有 个物品,要把其中一些物品装进容量为 的背包;
- 第 个物品自己的体积是 ;
- 如果把第 个物品放进背包,会获得 的收益;
- 要求在不超过容量的情况下,最大化收益。
01 背包就是这个物品要么选,要么不选。
这是模板。
实现
这个问题初学者较容易设计出错误的状态,如下:
- 表示前 个物品取得的最大价值。然而前面的物品占用了空间,所以会影响我现在决定能不能放第 个物品。有后效性。
- 表示占用 的空间取得的价值。然而每个物品只能用一次,决策的时候必须考虑哪些物品用了、哪些没用。有后效性。
那么我们就设计状态: 表示“只考虑前 个物品,恰好用 的容量能装下的最大价值”。这个状态没有后效性。
转移方程:
初始 ,目标(注意由于状态的设计,必须要取最大值,我们定义的是”恰好“,不是”至多“):。
之所以状态不定义为“至多用 的容量”,是因为这样无法确定还剩多少空间(但实际上这么做是可行的,具体方法请读者自行思考)。
memset(dp, 0xcf, sizeof(dp)); // -INF,因为要搞最大值
dp[0][0] = 0;
for (int i = 1; i <= n; ++i)
{
for (int j = 0; j <= V; ++j)
{
dp[i][j] = dp[i-1][j];
if (j >= v[i]) dp[i][j] = max(dp[i][j], dp[i-1][j-v[i]] + w[i]);
}
}
int ans = -1;
for (int i = 0; i <= V; ++i)
ans = max(ans, dp[n][i]);
注意到一个问题, 的循环次序是无关紧要的,这不会影响结果。
实际上,笔者目前没有找到任何能证明不用取最大值是错误的反例,也许这样的反例根本不存在,因为在更新时,由于体积更大,肯定能成功更新。实际上这个代码实现是将那些用不掉的体积填进去了空气,例如我们看这样一组数据:
n = 2, V = 4
物品 A: v = 3, w = 10
物品 B: v = 4, w = 1
它的 DP 数组长这样:
j= 0 1 2 3 4
i = 1: 0 0 0 10 10
i = 2: 0 0 0 10 10
这个 dp[1][4] = 10
从哪来的?显然,它是从 dp[1][1]
更新来的,我们填进去了体积为 的空气。
但考虑到实际中是不在意这点时间(取最大值才线性),且严格符合状态转移的过程,强烈建议取最大值。接下来的局部代码为了不影响阅读均不会取最大值,但是完整题目的代码均会取最大值。
滚动数组
更奇妙的是,还可以把 数组变成一维的。
for (int i = 1; i <= n; ++i)
{
scanf("%d%d", &v, &w);
for (int j = V; j >= v; --j)
dp[j] = max(dp[j], dp[j-v] + w);
}
为什么这样是对的?在计算 前, 里保存的是 ,而 保存的是 (注意 的循环次序),然后 可以顺利更新成 的内容,覆盖掉的内容再也不会使用,倒序的循环顺序可以使得计算 时 还未被计算,保存的是 。所以这样是可以的!
一个 DP 可以通过滚动数组来优化,当且仅当其状态图是分层的,下 层的结果,由上 层结果唯一确定。
在 01 背包中,第 层的结果由第 层唯一确定,因此可以使用滚动数组。而且由于顺序的特殊性,所以只用了一维。
但是如果没有这种特殊性怎么办?很简单:
int dp[2][MAXV];
for (int i = 1; i <= n; ++i)
{
for (int j = 0; j <= V; ++j)
{
dp[i & 1][j] = dp[(i-1) & 1][j];
if (j >= v[i]) dp[i & 1][j] = max(dp[i & 1][j], dp[(i-1) & 1][j-v[i]] + w[i]);
}
}
看到了吗?这样就与循环顺序没有关系了,让 DP 在两维中交替进行。
解的打印
使用滚动数组后,解的打印几乎变成天方夜谭了。事实上即使不用滚动数组,想打印解还是很困难(试一试会发现无法打印字典序最大的解),此时应该使用 表示“只考虑后 个物品,恰好用 的容量能装下的最大价值”。方法相同,打印时使用回溯法打印即可。实际上要求输出解的背包问题很少,但为了防住毒瘤可以考虑学习。
完全背包
这类问题的形式是:
- 你有 种物品,每种有无限多,要把其中一些物品装进容量为 的背包;
- 第 种物品自己的体积是 ;
- 如果把第 个物品放进背包,会获得 的收益;
- 要求在不超过容量的情况下,最大化收益。
这是模板。
根据经验,设 表示选前 种物品,体积为 ,物品最大价值,则有:
初始 ,目标:。
代码如下:
for (int i = 1; i <= n; ++i)
{
int v = read(), w = read();
for (int j = 0; j <= V; ++j)
f[i][j] = f[i-1][j];
// 这是另一种初始化的写法,代码较长但是效率较高
for (int j = v; j <= V; ++j)
f[i][j] = max(f[i][j], f[i][j-v] + w);
}
还是计算顺序的原因,在计算 时, 保存的是 ,而 保存的是 ,计算 时, 已经计算完毕。滚动数组代码如下:
for (int i = 1; i <= m; ++i)
{
int v = read(), w = read();
for (int j = v; j <= V; ++j)
f[j] = max(f[j], f[j-v] +w);
}
多重背包
这类问题的形式是:
- 你有 种物品,每种有 个,要把其中一些物品装进容量为 的背包;
- 第 种物品自己的体积是 ;
- 如果把第 个物品放进背包,会获得 的收益;
- 要求在不超过容量的情况下,最大化收益。
如果我们把每种物品都拆成 个物品,这几乎就跟 01 背包一模一样了。但遗憾的是,这样做很慢,时间复杂度高达 。
面对这一情况有两种做法:二进制拆分法和单调队列优化法,时间复杂度分别为 和 。其中单调队列优化法我们会在后续的动态规划教程中进行讨论,也可以阅读背包九讲来学习。
二进制拆分法基于这样一个事实:从 这 个整数中选出若干个相加,可以得到 之间的所有数。为什么?想一想计算机是怎么用二进制表示数的吧!
进一步地,我们可以求出满足 的最大整数 ,设 ,那么:
- 根据 的最大性,有 ,将 代入得 ,可推出 (两边同时减去 ),所以 ,因此从 中选出若干个相加可以得到 之间的任何整数(实际上是可以得到 之间的任意整数)。
- 从 以及 中选择若干个数相加(注意多了个 ),可以得到 之间的任何整数(就是把 中最后的结论用不等式性质一变形),而由于 的定义,加上数学结论 ,可得到 ,所以等量代换后可以证明从 以及 中选择若干个数相加,可以得到 之间的任何整数。而又有从 中选出若干个相加可以得到 之间的任何整数,因此从 以及 中选择若干个数相加,可以得到 之间的任何整数。
综上所述,我们只需要把数量为 的第 种物品拆成 个物品,体积为 ,它们可以凑成 之间所有能被 整除的数。
模板,代码如下:
查看代码
// 注:我们通常用 v 代表体积,w 代表价值,但是此题中是反的。笔者在编码时遵照的时常规而不是原题。
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
inline int read(void)
{
int x = 0, c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar();
return x;
}
int n, V;
int f[40005];
struct BadCreeper // (
{
int v, w;
BadCreeper(int v = 0, int w = 0) :
v(v), w(w) {}
};
int main(void)
{
n = read(), V = read();
for (int i = 1; i <= n; ++i)
{
int w = read(), v = read(), m = read(), c = 1;
vector <BadCreeper> a;
while (m - c >= 0)
{
m -= c;
a.push_back(BadCreeper(c * v, c * w));
c <<= 1;
}
if (m != 0) a.push_back(BadCreeper(m * v, m * w)); // R = 0,没必要添加
for (int j = 0; j < a.size(); ++j) // 枚举物品
for (int k = V; k >= a[j].v; --k) // 01 转移
f[k] = max(f[k], f[k - a[j].v] + a[j].w);
}
int ans = -1;
for (int i = 0; i <= V; ++i)
ans = max(ans, f[i]);
printf("%d\n", ans);
return 0;
}
混合背包
其实就是将前面三种背包混起来一起乱炖,有的物品只能取 次,有的能取 次,有的能取无限次。这怎么办?一种方法是将无限次化为最多能取得次数,这样就转化成多重背包了。但实际上没必要,我们只需要判断每种情况,然后套用不同种背包对应的循环即可。
模板,代码如下:
查看代码
// 注:此代码编写时仍按照标准背包的定义
#include <iostream>
#include <cstdio>
#include <string>
#include <vector>
#define pii pair<int, int>
#define X first
#define Y second
using namespace std;
inline int read(void)
{
int x = 0, c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar();
return x;
}
int n, V;
inline int calc(string s)
{
pii res;
bool flag = 0;
for (int i = 0; i < s.length(); ++i)
{
if (s[i] == ':')
{
flag = 1;
continue;
}
if (flag) res.Y = res.Y * 10 + (s[i] - '0');
else res.X = res.X * 10 + (s[i] - '0');
}
return res.X * 60 + res.Y;
}
int f[1005];
int main(void)
{
string B, E;
cin >> B >> E;
V = calc(E) - calc(B);
n = read();
for (int i = 1; i <= n; ++i)
{
int v = read(), w = read(), p = read();
if (p == 0)
{
for (int j = v; j <= V; ++j)
f[j] = max(f[j], f[j - v] + w);
}
else // 你也可以把 01 背包的情况单独扔出来,不过没必要
{
vector <pii> a;
int s = 1;
while (p > s)
{
p -= s;
a.push_back(make_pair(v * s, w * s));
s <<= 1;
}
a.push_back(make_pair(v * p, w * p));
for (int j = 0; j < a.size(); ++j)
for (int k = V; k >= a[j].X; --k)
f[k] = max(f[k], f[k - a[j].X] + a[j].Y);
}
}
int ans = 0;
for (int i = 0; i <= V; ++i)
ans = max(ans, f[i]);
printf("%d\n", ans);
return 0;
}
二维费用背包
二位费用背包是一类一个物品会消耗两种价值(比如汽水就是一种消耗金钱和寿命的东西)。
设 表示只考虑前 瓶汽水,消耗的金钱恰好为 ,消耗的阳寿恰好为 ,获得的脂肪是 ,那么转移方程便是:
同样可以使用滚动数组优化空间(只有第一维是阶段,剩下的是状态,不能滚),注意相应的循环顺序。
模板,代码如下:
查看代码
#include <iostream>
#include <cstdio>
using namespace std;
inline int read(void)
{
int x = 0, c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar();
return x;
}
int n, M, T;
int f[205][205];
int main(void)
{
n = read(), M = read(), T = read();
for (int i = 1; i <= n; ++i)
{
int m = read(), t = read();
for (int j = M; j >= m; --j)
for (int k = T; k >= t; --k)
f[j][k] = max(f[j][k], f[j-m][k-t] + 1); // 每个愿望带来的价值都是相等的,都是单位 1
}
int ans = 0;
for (int i = 0; i <= M; ++i)
for (int j = 0; j <= T; ++j)
ans = max(ans, f[i][j]);
cout << ans << endl;
return 0;
}
也很容易转化到二维完全背包、多重背包等问题,这里不做赘述。
分组背包
这类问题跟 01 背包的本质区别是:物品可以分为 组,每组中的物品相互冲突,即每组中最多只能选一个物品。问最大价值。模板。
设 表示在前 组中选出总体积为 的物品放入背包,物品的最大价值和。那么:
同样可以使用滚动数组滚掉第一维,代码如下:
查看代码
#include <iostream>
#include <cstdio>
#include <vector>
#define pii pair<int, int>
#define X first
#define Y second
using namespace std;
inline int read(void)
{
int x = 0, c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar();
return x;
}
int n, V, f[1005];
vector <pii> a[1005];
int main(void)
{
V = read(), n = read();
for (int i = 1; i <= n; ++i)
{
int v = read(), w = read(), c = read();
a[c].push_back(make_pair(v, w));
}
for (int i = 0; i < 1000; ++i)
{
if (a[i].size() == 0) continue; // 这组没有东西就不算了
for (int j = V; j >= 0; --j)
for (int k = 0; k < a[i].size(); ++k) // 决策
if (j >= a[i][k].X) f[j] = max(f[j], f[j - a[i][k].X] + a[i][k].Y);
}
int ans = 0;
for (int i = 0; i <= V; ++i)
ans = max(ans, f[i]);
printf("%d\n", ans);
return 0;
}
注意循环顺序,因为 i,j
是状态,k
是决策,顺序不能混淆。
依赖性背包
这类问题长这样:在 01 背包的基础上,某些物品依赖于一个主物品,比如你想购买 Minecraft Java Edition,你就需要注册 Microsoft 账号,否则是不能购买 Minecraft Java Edition 的。
这里我们讨论基础的依赖性背包,见模板。
我们对这个题做一个扩展:每个主件可以拥有无限个附件,这怎么办呢?可以将每个主件和它的附件作为分组背包中的一个组,想要对这个组进行背包问题的求解,就必须选择主件。对于包含一个主件和若干个附件的集合有以下可能性:仅选择主件,选择主件后再选择一个附件,选择主件后再选择两个附件……需要将以上可能性的容量和价值转换成一件件物品。因为这几种可能性只能选一种,所以可以将这看成分组背包。虽然这样做很慢,但足以通过原题(毕竟最多才两个附件)。对于本题而言,只需要依次判断每种情况即可。
代码如下:
查看代码
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
inline int read(void)
{
int x = 0, c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar();
return x;
}
struct BadCreeper
{
int v, w;
BadCreeper(int v = 0, int w = 0) :
v(v), w(w) {}
};
int n, V, f[32005];
vector <BadCreeper> a[65];
int main(void)
{
V = read(), n = read();
for (int i = 0; i < n; ++i)
{
int v = read(), p = read(), q = read();
if (q == 0) a[i].push_back(BadCreeper(v, v * p));
else a[q - 1].push_back(BadCreeper(v, v * p));
}
for (int i = 0; i < n; ++i)
for (int j = V; j >= 0; --j)
{
if (a[i].size() > 0 && j >= a[i][0].v)
f[j] = max(f[j], f[j - a[i][0].v] + a[i][0].w);
if (a[i].size() > 1 && j >= a[i][0].v + a[i][1].v)
f[j] = max(f[j], f[j - a[i][0].v - a[i][1].v] + a[i][0].w + a[i][1].w);
if (a[i].size() > 2)
{
if (j >= a[i][0].v + a[i][2].v)
f[j] = max(f[j], f[j - a[i][0].v - a[i][2].v] + a[i][0].w + a[i][2].w);
if (j >= a[i][0].v + a[i][1].v + a[i][2].v)
f[j] = max(f[j], f[j - a[i][0].v - a[i][1].v - a[i][2].v] + a[i][0].w + a[i][1].w + a[i][2].w);
}
}
int ans = 0;
for (int i = 0; i <= V; ++i)
ans = max(ans, f[i]);
printf("%d\n", ans);
return 0;
}
这类问题的更复杂形态我们会在树形 DP 中讨论。
Problemset
这里是一些简单各类背包问题的合集。它们本身不难,主要是帮助读者进一步了解背包,为接下来更困难的 DP 学习做准备。强烈读者仔细阅读这一部分,其中会出现一些前面的专题没有讲述的内容。
简单问题
这里的问题都比较简单,可以当作复习的题目。
[NOIP2001 普及组] 装箱问题
由于我们希望剩余空间尽可能地小,那么可以将物品的体积视作物品的价值,那么本题就是标准的 01 背包问题。代码如下:
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int v, n, w;
int f[30005];
int main(void)
{
scanf("%d%d", &v, &n);
for (int i = 1; i <= n; i++)
{
scanf("%d", &w);
for (int j = v; j >= w; j--)
f[j] = max(f[j], f[j - w] + w);
}
printf("%d", v - f[v]); // 根据之前的推论,这么做也可
return 0;
}
[CH5201] 数字组合
这个问的是将背包装满的方案总数。设计状态: 表示用前 个物品,恰好填满 的体积的方案总数。首先 必定为 。转移:,跟 01 背包很像?的确如此,对于这类改变问法的问题,一般只需将状态转移方程中的 max 改成 sum 即可。事实上,这样做可行的原因在于状态转移方程已经考察了所有可能的背包组成方案。
#include <iostream>
#include <cstdio>
using namespace std;
int n, m;
int f[10005];
int main(void)
{
scanf("%d%d", &n, &m);
f[0] = 1;
for (int i = 1; i <= n; ++i)
{
int w;
scanf("%d", &w);
for (int j = m; j >= w; --j) // 数字只能使用一次,对应 01 背包的顺序
f[j] += f[j - w];
}
printf("%d\n", f[m]);
return 0;
}
But WAIT! 这样不会产生之前所说的“不用取最大值也是对的”的填进去空气的情况吗?不会,我们只让 为 ,而 这种空气是 ,没有方案。
[CH5202] 自然数拆分 Lunatic 版
就是求方案总数的完全背包:容量为 ,有 的物品体积。注意要使用若干个,不能是 个(我拆成我自己),所以最终结果要 -1
。
查看代码
#include <iostream>
#include <cstdio>
using namespace std;
const unsigned int MOD = 2147483648u;
int n;
unsigned int f[4005];
int main(void)
{
scanf("%d", &n);
f[0] = 1;
for (int i = 1; i <= n; ++i)
for (int j = i; j <= n; ++j)
f[j] = (f[j - i] + f[j]) % MOD;
printf("%u\n", (f[n] - 1 + MOD) % MOD);
return 0;
}
[NOI Online #3 入门组] 买表
直接使用多重背包的模板即可。
查看代码
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int n, m;
bool f[500005];
int main(void)
{
scanf("%d%d", &n, &m);
f[0] = true;
for (int i = 1; i <= n; ++i)
{
int k, a, c = 1;
scanf("%d%d", &k, &a);
while (a - c >= 0)
{
a -= c;
int v = k * c;
for (int j = 500000; j >= v; --j)
f[j] |= f[j - v];
c <<= 1;
}
}
while (m--)
{
int x;
scanf("%d", &x);
puts(f[x] ? "Yes" : "No");
}
return 0;
}
[BJOI2019] 排兵布阵
假设 s=1
,那么这就是普通的背包问题了。现在将每个城堡排序一下,枚举要打过多少个玩家,就可以轻松算出贡献了。
查看代码
// 甚至连滚动数组都不用
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int s, n, m;
int a[105][105];
int f[105][20005]; // 只考虑前 i 座城堡,恰好使用 j 位士兵所获得的最大收益
int main(void)
{
scanf("%d%d%d", &s, &n, &m);
for (int i = 1; i <= s; ++i)
for (int j = 1; j <= n; ++j)
scanf("%d", &a[j][i]);
for (int i = 1; i <= n; ++i)
sort(a[i] + 1, a[i] + s + 1);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j) // 若使用滚动数组,需要倒序
{
f[i][j] = f[i - 1][j];
for (int k = 1; k <= s; ++k)
if (j >= a[i][k] * 2 + 1) f[i][j] = max(f[i][j], f[i - 1][j - a[i][k] * 2 - 1] + k * i); // 贡献为 k * i
}
int ans = 0;
for (int j = 0; j <= m; ++j)
ans = max(ans, f[n][j]);
printf("%d\n", ans);
return 0;
}
较复杂题目
这些题目会比较复杂。
[NOIP2018 提高组] 货币系统
大凯的疑惑。
可以猜测 的所有货币面额在 中一定都有,我们只需要看哪些没有。先排序,然后扫描,使用完全背包,可以拼成的就赋值为 。
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n;
int a[105];
bool f[25005];
int main(void)
{
int T;
cin >> T;
while (T--)
{
scanf("%d", &n);
for (int i = 1; i <= n; ++i) scanf("%d", a + i);
sort(a + 1, a + n + 1);
int ans = n;
memset(f, 0, sizeof(f));
f[0] = true;
for (int i = 1; i <= n; ++i)
{
if (f[a[i]])
{
--ans;
continue;
}
for (int j = a[i]; j <= 25000; ++j)
f[j] |= f[j - a[i]];
}
printf("%d\n", ans);
}
return 0;
}
[CSP-J2019] 纪念品
由于当日购买的纪念品也可以当日卖出换回金币,那么我们可以强制让第 天买的纪念品在第 天卖掉。但如果它应该在 天卖掉怎么办?不要紧,在第 天再买回来,价格是一样的。
于是直接完全背包板子拍上去就行了。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int t, n, m;
int p[105][105];
int f[10005];
int main(void)
{
scanf("%d%d%d", &t, &n, &m);
for (int i = 1; i <= t; ++i)
for (int j = 1; j <= n; ++j)
scanf("%d", p[i] + j);
for (int i = 1; i < t; ++i)
{
memset(f, 0, sizeof(f));
for (int j = 1; j <= n; ++j)
for (int k = p[i][j]; k <= m; ++k)
f[k] = max(f[k], f[k - p[i][j]] + p[i + 1][j] - p[i][j]);
m += f[m];
}
printf("%d\n", m);
return 0;
}
[Luogu P1441] 砝码称重
使用搜索枚举删掉哪些砝码,然后对剩下的砝码使用背包。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, m, sum = 0, ans = 0;
int a[25];
bool f[2005], del[25];
void dfs(int o, int cnt)
{
if (cnt == m)
{
memset(f, 0, sizeof(f));
f[0] = true;
int ret = 0;
for (int i = 1; i <= n; ++i)
if (!del[i])
{
for (int j = 2000; j >= 0; --j)
if (f[j] && !f[j + a[i]]) f[j + a[i]] = true, ++ret;
}
ans = max(ans, ret);
return;
}
if (o > n || cnt > m) return;
dfs(o + 1, cnt);
del[o] = true;
dfs(o + 1, cnt + 1);
del[o] = false;
}
int main(void)
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
{
scanf("%d", a + i);
sum += a[i];
}
dfs(1, 0);
printf("%d\n", ans);
return 0;
}
[Luogu P1782] 旅行商的背包
就是将多重背包和分组背包混在了一起(将奇货拆开),但是本题时间很紧,需要大力卡常。
查看代码
#include <iostream>
#include <cstdio>
#define V first
#define W second
using namespace std;
typedef long long i64;
inline int read(void)
{
int x = 0, c = getchar_unlocked(), f = 1;
while (!isdigit(c)) {if (c == '-') f = -1; c = getchar_unlocked();}
while (isdigit(c)) x = (x<<3) + (x<<1) + (c^48), c = getchar_unlocked();
return x * f;
}
int n, m, C, tot = 0;
i64 f[100005];
pair<i64, i64> a[2000005];
int main(void)
{
n = read(), m = read(), C = read();
while (n--)
{
i64 v = read(), w = read(), d = read(), c = 1;
// 这个二进制优化非常的强力,先处理物品再统一计算
while (d)
{
if (d >= c) d -= c;
else c = d, d = 0;
a[++tot].V = v * c, a[tot].W = w * c;
c <<= 1;
}
}
// 放到外面统一计算
for (int i = 1; i <= tot; ++i)
for (int j = C; j >= a[i].V; --j)
f[j] = max(f[j], f[j - a[i].V] + a[i].W);
while (m--)
{
i64 a = read(), b = read(), c = read();
for (int j = C; j >= 0; --j)
for (int v = 0; v <= j; ++v)
f[j] = max(f[j], f[j - v] + a * v * v + b * v + c);
}
printf("%lld\n", f[C]);
return 0;
}
[Luogu P1284] 三角形牧场
注意到所有的木板都需要使用,那么我们只需要确定三角形两条边的长度就可以知道第三边,然后使用海伦公式计算三角形面积。只需要配合一个二维状态的 01 背包就可以计算这样的长度是否可以得到满足。
查看代码
#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
double S(double a, double b, double c)
{
double p = (a + b + c) / 2;
return sqrt(p * (p - a) * (p - b) * (p - c));
}
bool check(int a, int b, int c)
{
if (a <= 0 || b <= 0 || c <= 0) return false;
if (a + b <= c || a + c <= b || b + c <= a) return false;
return true;
}
int n;
int l[45];
bool f[805][805];
int main(void)
{
scanf("%d", &n);
int sum = 0;
for (int i = 1; i <= n; ++i)
scanf("%d", l + i), sum += l[i];
f[0][0] = true;
for (int i = 1; i <= n; ++i)
for (int j = 800; j >= 0; --j)
for (int k = 800; k >= 0; --k)
{
if (j - l[i] >= 0) f[j][k] |= f[j - l[i]][k];
if (k - l[i] >= 0) f[j][k] |= f[j][k - l[i]];
}
double ans = -1.0;
for (int i = 0; i <= 800; ++i)
for (int j = 0; j <= 800; ++j)
if (f[i][j] && check(i, j, sum - i - j)) ans = max(ans, S(i, j, sum - i - j));
if (ans == -1.0) puts("-1");
else printf("%d\n", int(ans * 100));
return 0;
}
[Luogu P1156] 垃圾陷阱
设 代表高度为 时的最长存活时间,然后就是 01 背包了(采用刷表法)。
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
struct Rubbish
{
int t, f, h;
bool operator < (const Rubbish &a) const
{
return t < a.t;
}
} a[105];
int D, G;
int f[105];
int main(void)
{
scanf("%d%d", &D, &G);
for (int i = 1; i <= G; ++i)
scanf("%d%d%d", &a[i].t, &a[i].f, &a[i].h);
sort(a + 1, a + G + 1);
f[0] = 10;
for (int i = 1; i <= G; ++i)
for (int j = D; j >= 0; --j)
if (f[j] >= a[i].t)
{
if (j + a[i].h >= D)
{
printf("%d\n", a[i].t);
return 0;
}
f[j + a[i].h] = max(f[j + a[i].h], f[j]);
f[j] += a[i].f;
}
printf("%d\n", f[0]);
return 0;
}
[USACO03FALL] Cow Exhibition G
考虑将智商设定为体积,情商设定为价值,答案计算的时候枚举智商。注意滚动的时候的转移顺序。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int E = 400000, C = 400000;
int n;
int f[900005];
int main(void)
{
scanf("%d", &n);
memset(f, 0xbf, sizeof(f));
f[E] = 0;
for (int i = 1; i <= n; ++i) {
int a, b;
scanf("%d%d", &a, &b);
if (a >= 0) {
for (int j = C + E; j >= a; --j)
f[j] = max(f[j], f[j - a] + b);
} else {
for (int j = 0; j <= C + E - a; ++j)
f[j] = max(f[j], f[j - a] + b);
}
}
int ans = 0;
for (int i = 0; i <= C; ++i)
if (f[i + E] >= 0) ans = max(ans, i + f[i + E]);
printf("%d\n", ans);
return 0;
}
[SCOI2009] 粉刷匠
windy 有 条木板需要被粉刷。每条木板被分为 个格子。每个格子要被刷成红色或蓝色。windy 每次粉刷,只能选择一条木板上一段连续的格子,然后涂上一种颜色。 每个格子最多只能被粉刷一次。如果 windy 只能粉刷 次,他最多能正确粉刷多少格子?一个格子如果未被粉刷或者被粉刷错颜色,就算错误粉刷。。
考虑使用分组背包,每一个组是每一个木板,组内的物品是不同的粉刷次数。物品的价值可以使用 DP 预处理。
查看代码
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int n, m, T;
vector<int> a[55];
int f[55][55][55][2]; // f(id,i,j,flag) 第 id 块木板,考虑到第 i 个,染了 j 次,第 i 块是否正常染色
int g[2505];
void work(int id) {
int siz = a[id].size() - 1;
for (int i = 1; i <= siz; ++i)
for (int j = siz; j >= 1; --j) {
f[id][i][j][0] = max(f[id][i - 1][j][0], f[id][i - 1][j][1]);
f[id][i][j][1] = max(f[id][i - 1][j - 1][0], f[id][i - 1][j - 1][1]) + a[id][i];
if (i >= 2) f[id][i][j][1] = max(f[id][i][j][1], f[id][i - 2][j][1] + a[id][i]);
}
}
int main(void)
{
scanf("%d%d%d", &n, &m, &T);
for (int i = 1; i <= n; ++i) {
a[i].push_back(0);
static char s[55];
scanf("%s", s + 1);
for (int j = 1; j <= m; ++j)
if (a[i].empty() || s[j] != s[j - 1]) a[i].push_back(1);
else ++a[i][a[i].size() - 1];
}
for (int id = 1; id <= n; ++id) work(id);
for (int i = 1; i <= n; ++i) {
int siz = a[i].size() - 1;
for (int j = T; j >= 0; --j)
for (int k = 0; k <= siz; ++k)
if (j >= k) g[j] = max(g[j], g[j - k] + max(f[i][siz][k][0], f[i][siz][k][1]));
}
printf("%d\n", g[T]);
return 0;
}
[HEOI2013] Eden 的新背包问题
多重背包,但是多组询问,每次给定体积,并删去一个物品。询问相互独立。
用前后缀合并的做法,分别处理分组背包即可。
查看代码
#include <iostream>
#include <cstdio>
#include <vector>
#define pii pair<int, int>
#define V first
#define W second
using namespace std;
int n, q;
int f1[1005][1005], f2[1005][1005];
vector<pii> a[1005];
int main(void)
{
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
int v, w, m, c = 1;
scanf("%d%d%d", &v, &w, &m);
while (m >= c) {
m -= c;
a[i].push_back({c * v, c * w});
c <<= 1;
}
if (m) a[i].push_back({m * v, m * w});
}
for (int i = 1; i <= n; ++i) {
for (int j = 0; j <= 1000; ++j) f1[i][j] = f1[i - 1][j];
for (pii x : a[i])
for (int j = 1000; j >= x.V; --j)
f1[i][j] = max(f1[i][j], f1[i][j - x.V] + x.W);
}
for (int i = n; i >= 1; --i) {
for (int j = 0; j <= 1000; ++j) f2[i][j] = f2[i + 1][j];
for (pii x : a[i])
for (int j = 1000; j >= x.V; --j)
f2[i][j] = max(f2[i][j], f2[i][j - x.V] + x.W);
}
scanf("%d", &q);
while (q--) {
int d, e;
scanf("%d%d", &d, &e); ++d;
int ans = 0;
for (int j = 0; j <= e; ++j)
ans = max(ans, f1[d - 1][j] + f2[d + 1][e - j]);
printf("%d\n", ans);
}
return 0;
}
[Luogu P4141] 消失之物
背包,每次去掉一个物品,问有多少种方式填满。
考虑每次转移的时候加上了什么,我们倒序加了,那么查询的时候正序减去就好了。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, m;
int a[2005], f[2005], g[2005];
int main(void)
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) scanf("%d", a + i);
f[0] = 1;
for (int i = 1; i <= n; ++i)
for (int j = m; j >= a[i]; --j)
f[j] = (f[j] + f[j - a[i]]) % 10;
for (int i = 1; i <= n; ++i) {
memcpy(g, f, sizeof(g));
for(int j = a[i]; j <= m; ++j)
g[j] = (f[j] - g[j - a[i]] + 10) % 10;
for (int j = 1; j <= m; ++j)
printf("%d", g[j]);
putchar('\n');
}
return 0;
}
小结
背包问题主要以算法模板的形式出现在考试中,难在一道题如何转化成背包问题,但这不是本文讲解的重点。想要学会这一点,只能通过学习各种各样的 DP,做大量的题目,积累经验。请读者务必记住各种背包的原理,而不是背住循环顺序。