区间 DP 是线性 DP 的一种特殊形式,它在分阶段地划分问题时,与阶段中元素出现的顺序和由前一阶段的哪些元素合并而来有很大的关系。高维 DP 则是线性 DP 的扩展,大部分是从一条链扩展为了二维结构。
这两个内容都属于都属于线性 DP 的变形。
区间 DP
区间类动态规划是线性动态规划的扩展。
它在分阶段地划分问题时,与阶段中元素出现的顺序和由前一阶段的哪些元素合并而来由很大的关系。其转移方程一般形如 , 为将这两组元素合并起来的代价。
也有另一类区间 DP 的转移方程形如 ,这一类的区间是通过左右端点逐个扩展的。
由于大区间的答案依赖于小区间的答案,所以要将区间的长度作为“阶段”来进行 DP。
但是这么说很闹鬼,我们通过题目来认识区间 DP。
模板
我们分别来看一看刚才所说的两种转移的具体实例。
[NOI1995] 石子合并
以最大得分为例。
先假设排成一排,设 表示合并区间 的最大得分,则有 。
编程时有两种思路,一是记忆化搜索,二是递推,但注意递推时要以区间长度递增的顺序来推,因为大区间的答案依赖于小区间的答案。
接下来考虑如何处理环。我们将这条链延长两倍,变成 堆,这样就可以转化成 条链,分别为 。这种复制二倍链是处理环形 DP 极为有效的方式(因为环形问题中肯定有一条边是用不到的,那么我们将它从任意位置断开,然后就相当于从每个位置断开的情况都求了)。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
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;
int a[205], sum[205];
int f[205][205];
int main(void)
{
n = read();
for (int i = 1; i <= n; ++i) a[i] = a[n + i] = read();
for (int i = 1; i <= (n << 1); ++i) sum[i] = sum[i-1] + a[i];
int ans = 1000000000;
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= (n << 1); ++i) f[i][i] = 0;
for (int len = 2; len <= n; ++len) // 我们只需要处理到 len = n 即可
for (int i = 1; i <= (n << 1) - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k)
f[i][j] = min(f[i][j], f[i][k] + f[k+1][j] + sum[j] - sum[i-1]);
}
for (int i = 1; i <= n; ++i) ans = min(ans, f[i][i + n - 1]);
printf("%d\n", ans);
ans = 0;
memset(f, 0, sizeof(f));
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= (n << 1) - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k)
f[i][j] = max(f[i][j], f[i][k] + f[k+1][j] + sum[j] - sum[i-1]);
}
for (int i = 1; i <= n; ++i) ans = max(ans, f[i][i + n - 1]);
printf("%d\n", ans);
return 0;
}
状态有 个,每个状态的决策有 个,因此总时间复杂度为 。
[IOI2000] 回文字串
设 表示处理区间 的最小代价,那么显然可以花费 来处理左端点或右端点,如果左右端点相等的话还可以等价于处理 。显然,这是第二种转移。
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n;
char s[1005];
int f[1005][1005];
int main(void)
{
scanf("%s", s + 1);
n = strlen(s + 1);
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= n; ++i)
f[i][i] = 0, f[i][i - 1] = 0;
for (int len = 2; len <= n; ++len)
{
for (int i = 1; i <= n - len + 1; ++i)
{
int j = i + len - 1;
f[i][j] = min(f[i][j - 1], f[i + 1][j]) + 1;
if (s[i] == s[j]) f[i][j] = min(f[i][j], f[i + 1][j - 1]);
}
}
printf("%d\n", f[1][n]);
return 0;
}
状态有 个,每个状态的决策只有有 个,因此总时间复杂度为 。
简单应用
这里是一些简单的区间 DP 题目。
[CQOI2007] 涂色
设 为完成 涂色需要的最少步数。首先有 。
当 时,(只需要在涂其中某一个时,先涂这种颜色,将涂色的区间改为 ,再考虑 即可)。
在任何情况下, 都能由 和 拼出来。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n;
char s[55];
int f[55][55];
int main(void)
{
scanf("%s", s + 1);
n = strlen(s + 1);
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= n; ++i)
f[i][i] = 1;
for (int len = 1; len < n; ++len)
for (int i = 1; i <= n - len; ++i) {
int j = i + len;
if (s[i] == s[j]) f[i][j] = min(f[i][j - 1], f[i + 1][j]);
for (int k = i; k < j; ++k)
f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j]);
}
printf("%d\n", f[1][n]);
return 0;
}
[NOIP2007 提高组] 矩阵取数游戏
每一行的操作都是独立的,因此可以一行一行的思考。设 为一行中取到只剩下 的最大值,转移只有两种方式:左取和右取。时间复杂度为 。注意需要高精(当然可以用 __int128
)。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define LL __int128
using namespace std;
inline LL read(void)
{
LL x = 0;
int c = getchar();
while (!isdigit(c)) c = getchar();
while (isdigit(c)) x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
return x;
}
inline void print(LL x)
{
if (x > 9) print(x / 10);
putchar(x % 10 + '0');
}
int n, m;
LL a[85];
LL f[85][85];
int main(void)
{
n = read(), m = read();
LL ans = 0;
while (n--) {
for (int i = 1; i <= m; ++i) a[i] = read();
memset(f, 0, sizeof(f));
f[1][m] = 0;
LL k = 1;
for (int len = m - 1; len >= 0; --len) {
k <<= 1;
for (int i = 1; i <= m - len + 1; ++i) {
int j = i + len - 1;
f[i][j] = max(f[i - 1][j] + a[i - 1] * k, f[i][j + 1] + a[j + 1] * k);
}
}
LL res = -1;
for (int i = 1; i <= m; ++i)
res = max(res, f[i][i] + a[i] * k);
ans += res;
}
print(ans); putchar('\n');
return 0;
}
[CF245H] Queries for Number of Palindromes
尽管用专门的字符串算法可以很高效地解决这个问题,但是还可以用区间 DP 来解决。根据经验,我们设 表示这个范围内的回文子串个数。那么根据容斥原理显然有:
现在的问题就是如何处理 is_pal
。虽然字符串 Hash 可以在线性复杂度内解决这个问题,但没必要,因为我们的 DP 是平方复杂度。实际上这个问题也可以以平方复杂度用区间 DP 来求解。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
char s[5005];
int n, f[5005][5005];
bool is_pal[5005][5005];
int main(void)
{
scanf("%s", s + 1);
n = strlen(s + 1);
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
if (s[i] == s[j] && (i + 1 >= j || is_pal[i + 1][j - 1])) is_pal[i][j] = true;
}
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
f[i][j] = f[i + 1][j] + f[i][j - 1] - f[i + 1][j - 1] + is_pal[i][j];
}
int q, l, r;
scanf("%d", &q);
while (q--) {
scanf("%d%d", &l, &r);
printf("%d\n", f[l][r]);
}
return 0;
}
[IOI1998] Polygon
Portal.
特别地,请将数据范围看作 。
我们可以枚举删哪一条边,本题就转化为链上问题。设 表示区间 的最大值行吗?不行!由于负负得正的存在,这样的状态设计不满足最优子结构性质,那怎么办?在记录一个记录最小值的就行了!
设 分别代表区间的最大最小值。当运算为求和是,正常更新即可。乘法需要注意,最大值的来源只可能是最大乘最大或最小乘最小,而最小值的来源则是所有的都可能。实际上,这时候如果搞不清楚,还不如都写上,反正不会得出错误的解就行(没人会卡这里的常数吧)。代码如下:
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int INF = 1000000000;
int n;
char c[105];
int a[105];
int f[105][105][2];
int main(void)
{
cin >> n;
for (int i = 1; i <= n; ++i)
cin >> c[i] >> a[i];
for (int i = n + 1; i <= (n << 1); ++i)
c[i] = c[i - n], a[i] = a[i - n];
for (int l = 1; l <= (n << 1); ++l)
for (int r = 1; r <= (n << 1); ++r)
if (l != r) f[l][r][0] = -INF, f[l][r][1] = INF;
else f[l][r][0] = f[l][r][1] = a[l];
for (int len = 2; len <= (n << 1); ++len)
for (int l = 1; l <= (n << 1) - len + 1; ++l) {
int r = l + len - 1;
for (int k = l; k < r; ++k) {
if (c[k + 1] == 't') {
f[l][r][0] = max(f[l][r][0], f[l][k][0] + f[k + 1][r][0]);
f[l][r][1] = min(f[l][r][1], f[l][k][1] + f[k + 1][r][1]);
} else {
f[l][r][0] = max({f[l][r][0], f[l][k][0] * f[k + 1][r][0], f[l][k][1] * f[k + 1][r][1]});
f[l][r][1] = min({f[l][r][1], f[l][k][0] * f[k + 1][r][1], f[l][k][1] * f[k + 1][r][0], f[l][k][0] * f[k + 1][r][0], f[l][k][1] * f[k + 1][r][1]});
}
}
}
int ans = -INF;
for (int i = 1; i <= n; ++i) ans = max(ans, f[i][i + n - 1][0]);
printf("%d\n", ans);
for (int i = 1; i <= n; ++i)
if (f[i][i + n - 1][0] == ans)
printf("%d ", i);
putchar('\n');
return 0;
}
[SCOI2003] 字符串折叠
首先这肯定是第一种区间 DP,因为折叠可以从中间抠,肯定不是从两头扩展。
设 为 的最短折叠,按照它的题意,应该有以下三种折叠方式:
- 自己作为一个折叠,显然这时 ;
- 把若干个重复的东西拼成一个,这怎么搞?不要紧,数据范围那么小,我们暴力点想,这些重复的东西长度肯定是一样的,那么我们就枚举长度 ,而且显然要求 ,这一过程我们
for
一遍检查是不是约数就可以了,因为检查能不能折叠的操作更费时间,加上括号和数字的代价,有 ,其中 表示求一个数字的位数,在这之前我们要线性扫一遍看能不能折叠,进行一轮这样的转移的时间复杂度为 ; - 枚举中间点进行转移,就是把两坨东西拼在一起,这时候 。
状态有 个,总时间复杂度为 。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, D[105];
char s[105];
int f[105][105];
inline bool check(int l, int r, int len)
{
for (int i = l; i <= r; ++i)
if (s[i] != s[(i - l) % len + l]) return false;
return true;
}
int main(void)
{
for (int i = 0; i < 10; ++i) D[i] = 1;
for (int i = 10; i < 100; ++i) D[i] = 2;
for (int i = 100; i < 103; ++i) D[i] = 3;
scanf("%s", s + 1);
n = strlen(s + 1);
for (int i = 1; i <= n; ++i)
for (int j = i; j <= n; ++j)
f[i][j] = j - i + 1;
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k)
f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j]);
for (int k = 1; k <= len; ++k) {
if (len % k != 0) continue;
if (check(i, j, k)) f[i][j] = min(f[i][j], f[i][i + k - 1] + 2 + D[len / k]);
}
}
printf("%d\n", f[1][n]);
return 0;
}
高维 DP
其实在之前就已经见识过高维 DP 了。像 [NOIP2002 普及组] 过河卒。要知道的是它们也可以用滚动数组优化。我们直接上题:
简单内容
一道简单题。
[NOIP2009 普及组] 道路游戏
设 代表第 个时刻获得的最大收益,转移的时候枚举机器人走的距离和从哪个位置开始走的即可(这样价值就可以计算出来了,用前缀和预处理加速查询)。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, m, p;
int a[1005][1005], s[1005][1005];
int f[1005], cost[1005];
int main(void)
{
scanf("%d%d%d", &n, &m, &p);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &a[j][i % n]);
for (int i = 1; i <= m; ++i)
for (int j = 0; j < n; ++j)
s[i][j] = s[i - 1][(j - 1 + n) % n] + a[i][j];
for (int i = 0; i < n; ++i)
scanf("%d", cost + i);
memset(f, 0xbf, sizeof(f));
f[0] = 0;
for (int i = 1; i <= m; ++i)
for (int j = 0; j < n; ++j)
for (int k = 1; k <= min(i, p); ++k)
f[i] = max(f[i], f[i - k] + s[i][j] - s[i - k][((j - k) % n + n) % n] - cost[((j - k) % n + n) % n]);
printf("%d\n", f[m]);
return 0;
}
普通内容
两道正常一点的题。
[NOIP2014 提高组] 飞扬的小鸟
设 代表飞到 ,高度为 的最小点击数。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
constexpr int INF = 0x3f3f3f3f;
int n, m, k;
int x[10005], y[10005], low[10005], high[10005];
bool e[10005];
int f[10005][2005];
int main(void)
{
scanf("%d%d%d", &n, &m, &k);
for (int i = 1; i <= n; ++i)
{
scanf("%d%d", x + i, y + i);
low[i] = 1, high[i] = m;
}
for (int i = 1, a, b, c; i <= k; ++i)
{
scanf("%d%d%d", &a, &b, &c);
e[a] = true;
low[a] = b + 1;
high[a] = c - 1;
}
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= m; ++i) f[0][i] = 0;
for (int i = 1; i <= n; ++i)
{
for (int j = x[i] + 1; j <= x[i] + m; ++j) f[i][j] = min(f[i - 1][j - x[i]] + 1, f[i][j - x[i]] + 1); // 上升
for (int j = m + 1; j <= x[i] + m; ++j) f[i][m] = min(f[i][m], f[i][j]); // 将飞到天上的拉回来
for (int j = 1; j <= m - y[i]; ++j) f[i][j] = min(f[i][j], f[i - 1][j + y[i]]); // 下降
// 禁区
for (int j = 0; j < low[i]; ++j) f[i][j] = INF;
for (int j = high[i] + 1; j <= m; ++j) f[i][j] = INF;
}
int ans = INF;
for (int i = 1; i <= n; ++i) ans = min(ans, f[n][i]);
if (ans < INF) printf("1\n%d\n", ans);
else
{
puts("0");
for (int i = n; i >= 1; --i)
{
bool flag = false;
for (int j = 1; j <= m; ++j)
if (f[i][j] < INF)
{
flag = true;
break;
}
if (flag)
{
ans = 0;
for (int j = 1; j <= i; ++j)
if (e[j]) ++ans;
printf("%d\n", ans);
return 0;
}
}
}
return 0;
}
[CSP-S2019] Emiya 家今天的饭
如果没有第三种限制,那么直接使用乘法原理碾过去即可。那么我们只需要求解出违反限制三的方案数,然后减去即可。
也就是说要求由食材的出现次数大于 ,不难发现满足这一条件的食材最多只有 个,否则 。于是我们可以枚举这一个食材,设该食材为 ,每一个食材都进行 DP:
设 代表考虑前 个烹饪方法,总共选出 道菜,其中 个菜使用食材 ,那么:
但是这样的状态数爆炸了,有 个,转移有 个,还要进行 次 DP,时间复杂度 ,无法通过。
意味着什么?注意它等价于 ,也就是 ,也就是使用 的菜肴数量减去未使用 的菜肴数量。设 代表考虑前 个烹饪方法, 用的次数减掉 未被用的次数为 ,那么:
- 若 ,则 ;
- 若 ,则 ;
- 不选, 。
最后的答案就是所有 为正的 和,这样就满足 。
还可以进一步进行优化:对于第二种转移,我们不需要枚举 ,而是直接使用 的前缀和减去 即可。
实现时由于数组的下标不能是负的,所以我们只需要给下标加上 即可。最终时间复杂度为 。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MOD = 998244353;
int n, m;
int S[105];
int a[105][2005];
int f[105][205];
int main(void)
{
scanf("%d%d", &n, &m);
int ans = 1;
for (int i = 1; i <= n; ++i)
{
for (int j = 1; j <= m; ++j)
scanf("%d", &a[i][j]), S[i] = (S[i] + a[i][j]) % MOD;
ans = 1ll * ans * (S[i] + 1) % MOD;
}
ans = (ans - 1 + MOD) % MOD;
for (int x = 1; x <= m; ++x)
{
memset(f, 0, sizeof(f));
f[0][0 + 100] = 1;
for (int i = 1; i <= n; ++i)
{
int re = (S[i] + MOD - a[i][x]) % MOD; // 利用前缀和进行计算
for (int j = 1; j <= 201; ++j)
{
f[i][j + 1] = (f[i][j + 1] + 1ll * a[i][x] * f[i - 1][j]) % MOD;
f[i][j - 1] = (f[i][j - 1] + 1ll * re * f[i - 1][j]) % MOD;
f[i][j] = (f[i][j] + f[i - 1][j]) % MOD;
}
}
int no = 0;
for (int i = 1; i <= n; ++i)
ans = (ans - f[n][i + 100] + MOD) % MOD;
}
printf("%d\n", ans);
return 0;
}
Problemset
嗯,题真多。
区间 DP
这里是区间 DP 的内容。
[CF607B] Zuma
设 代表消掉区间 的答案,转移对于读到这里的读者来说不是困难。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n;
int a[505];
int f[505][505];
int main(void)
{
scanf("%d", &n);
for (int i = 1; i <= n; ++i) scanf("%d", a + i);
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= n; ++i) f[i][i] = 1, f[i][i - 1] = 1;
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i)
{
int j = i + len - 1;
if (a[i] == a[j]) f[i][j] = f[i + 1][j - 1];
for (int k = i; k < j; ++k)
f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j]);
}
printf("%d\n", f[1][n]);
return 0;
}
[NOIP2003 提高组] 加分二叉树
设 代表区间 的最大答案,那么 。初始条件要注意,由于空树的贡献是 ,所以要令 。本题需要打印解,用 记录区间 所选取的根节点,然后递归遍历二叉树即可。
查看代码
#include <iostream>
#include <cstdio>
#define i64 long long
using namespace std;
int n;
int a[35], root[35][35];
i64 f[35][35];
void print_ans(int l, int r)
{
if (l > r) return;
printf("%d ", root[l][r]);
if (l == r) return;
print_ans(l, root[l][r] - 1);
print_ans(root[l][r] + 1, r);
}
int main(void)
{
scanf("%d", &n);
for (int i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
f[i][i] = a[i];
f[i][i - 1] = 1;
root[i][i] = i;
}
f[n + 1][n] = 1;
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i)
{
int j = i + len - 1;
for (int k = i; k < j; ++k)
if (f[i][j] < f[i][k - 1] * f[k + 1][j] + a[k])
{
f[i][j] = f[i][k - 1] * f[k + 1][j] + a[k];
root[i][j] = k;
}
}
printf("%lld\n", f[1][n]);
print_ans(1, n);
putchar('\n');
return 0;
}
[NOIP2006 提高组] 能量项链
环上问题肯定要复制成二倍链,而且为了方便计算我们最好现予处理出能量珠的头尾标记。转移时 。相乘的三个数分别是左区间对应的头标记,右区间对应的头标记和右区间对应的尾标记。
查看代码
#include <iostream>
#include <cstdio>
using namespace std;
int n;
int a[205], head[205], tail[205];
int f[205][205];
int main(void)
{
scanf("%d", &n);
for (int i = 1; i <= n; ++i)
{
scanf("%d", a + i);
head[i] = head[i + n] = a[i];
tail[i - 1] = tail[i + n - 1] = a[i];
}
tail[n << 1] = a[1];
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= (n << 1) - len + 1; ++i)
{
int j = i + len - 1;
for (int k = i; k < j; ++k)
f[i][j] = max(f[i][j], f[i][k] + f[k + 1][j] + head[i] * head[k + 1] * tail[j]);
}
int ans = 0;
for (int i = 1; i <= n; ++i)
ans = max(ans, f[i][i + n - 1]);
printf("%d\n", ans);
return 0;
}
[Luogu P1220] 关路灯
设 表示关掉 的最小功率?不行,因为我们还需要知道张大爷(不是笔者)的位置,到底是在 还是在 。所以 表示张大爷在 , 表示张大爷在 。转移对读到这的读者来说不是困难。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
int n, c;
int p[155], w[155];
int sum[155];
int f[155][155][2];
int calc(int x, int y) { // 除开区间 [x, y] 的和
return sum[x - 1] + sum[n] - sum[y];
}
int main(void)
{
memset(f, 0x3f, sizeof(f));
scanf("%d%d", &n, &c);
for (int i = 1; i <= n; ++i) {
scanf("%d%d", p + i, w + i);
sum[i] = sum[i - 1] + w[i];
}
f[c][c][0] = f[c][c][1] = 0;
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
f[i][j][0] = min(f[i + 1][j][0] + (p[i + 1] - p[i]) * calc(i + 1, j), f[i + 1][j][1] + (p[j] - p[i]) * calc(i + 1, j));
f[i][j][1] = min(f[i][j - 1][0] + (p[j] - p[i]) * calc(i, j - 1), f[i][j - 1][1] + (p[j] - p[j - 1]) * calc(i, j - 1));
}
printf("%d\n", min(f[1][n][0], f[1][n][1]));
return 0;
}
[USACO07OPEN] Cheapest Palindrome G
设 代表完成 的最小代价,然后分类转移。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, m;
char s[2005];
int f[2005][2005];
int a[30], b[30];
int main(void)
{
cin >> n >> m;
scanf("%s", s + 1);
for (int i = 1; i <= n; ++i) {
char x; int y, z;
cin >> x >> y >> z;
a[x - 'a'] = y, b[x - 'a'] = z;
}
memset(f, 0x3f, sizeof(f));
for (int i = 1; i <= m; ++i)
f[i][i] = 0, f[i][i - 1] = 0;
f[m + 1][m] = 0;
for (int len = 2; len <= m; ++len)
for (int i = 1; i <= m - len + 1; ++i) {
int j = i + len - 1;
if (s[i] == s[j]) f[i][j] = min(f[i][j], f[i + 1][j - 1]);
f[i][j] = min(f[i][j], f[i][j - 1] + min(b[s[j] - 'a'], a[s[j] - 'a']));
f[i][j] = min(f[i][j], f[i + 1][j] + min(b[s[i] - 'a'], a[s[i] - 'a']));
}
printf("%d\n", f[1][m]);
return 0;
}
[HNOI2010] 合唱队
设 表示形成理想队列中区间 的方案数?但是这样我们不知道上一个人是谁,那么就设 代表上一个人从左边加, 代表上一个人从右边加,初始条件仅设其中一个为 即可(否则会算重)。
查看代码
#include <bits/stc++.h>
using namespace std;
const int MOD = 19650827;
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;
int a[1005];
int f[1005][1005][2];
int main(void) {
n = read();
for (int i = 1; i <= n; ++i) {
a[i] = read();
f[i][i][0] = 1;
}
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
if (a[i] < a[i + 1]) f[i][j][0] += f[i + 1][j][0];
if (a[i] < a[j]) f[i][j][0] += f[i + 1][j][1];
f[i][j][0] %= MOD;
if (a[j] > a[i]) f[i][j][1] += f[i][j - 1][0];
if (a[j] > a[j - 1]) f[i][j][1] += f[i][j - 1][1];
f[i][j][1] %= MOD;
}
cout << (f[1][n][0] + f[1][n][1]) % MOD << endl;
return 0;
}
[UVA1336] Fixing the Great Wall
和“关路灯”很相似。这种不知到当前的时间,而在转移时将“未来增加的费用”计算好,然后“时间归零”的技巧很有用。
查看代码
#include <bits/stdc++.h>
using namespace std;
int n, x;
double v;
double f[1005][1005][2]; // k = 0 在 i, k = 1 在 j
int s[1005];
struct mouse {
int x, c, d;
mouse(int x = 0, int c = 0, int d = 0) :
x(x), c(c), d(d) {}
bool operator < (const mouse &a) const {
return x < a.x;
}
} a[1005];
int g(int x, int y) {
return s[n] - (s[y] - s[x - 1]);
}
int main(void)
{
// 计算未发生,但是肯定会发生的代价,然后“时钟归零”
while (scanf("%d%lf%d", &n, &v, &x) == 3 && n) {
for (int i = 1; i <= n; ++i)
scanf("%d%d%d", &a[i].x, &a[i].c, &a[i].d);
a[++n] = mouse(x);
sort(a + 1, a + n + 1);
int t, sum = 0;
for (int i = 1; i <= n; ++i) {
if (a[i].x == x) t = i;
s[i] = s[i - 1] + a[i].d;
sum += a[i].c;
}
for (int i = 1; i <= n; ++i)
for (int j = i; j <= n; ++j)
f[i][j][0] = f[i][j][1] = 1e9;
f[t][t][0] = f[t][t][1] = 0;
for (int len = 2; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
f[i][j][0] = min(f[i + 1][j][0] + (a[i + 1].x - a[i].x) / v * g(i + 1, j),
f[i + 1][j][1] + (a[j].x - a[i].x) / v * g(i + 1, j));
f[i][j][1] = min(f[i][j - 1][0] + (a[j].x - a[i].x) / v * g(i, j - 1),
f[i][j - 1][1] + (a[j].x - a[j - 1].x) / v * g(i, j - 1));
}
printf("%d\n", int(sum + min(f[1][n][0], f[1][n][1])));
}
return 0;
}
[USACO19DEC] Greedy Pie Eaters P
考虑计算 为吃掉 的派所能得到的最大体重。那么有 ,其中 代表区间 中 未被吃掉,现在吃了获得的最大体重,可以通过区间 DP 预处理得到。
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int n, m;
int w[90005], l[50005], r[50005];
int f[305][305], g[305][305][305];
int main(void)
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= m; ++i) {
int w, l, r;
scanf("%d%d%d", &w, &l, &r);
for (int j = l; j <= r; ++j)
g[l][r][j] = w;
}
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k <= j; ++k)
g[i][j][k] = max({g[i][j][k], g[i + 1][j][k], g[i][j - 1][k]});
}
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k)
f[i][j] = max(f[i][j], f[i][k] + f[k + 1][j]);
for (int k = i; k <= j; ++k)
f[i][j] = max(f[i][j], f[i][k - 1] + f[k + 1][j] + g[i][j][k]);
}
printf("%d\n", f[1][n]);
return 0;
}
[CSP-S 2021] 括号序列
大型分类讨论,注意不要算重,有许多种方法可以解决这个问题:细分状态或者对特别处理算重的情况。对于读到这里的读者来说应该能完成这道题。
查看代码
#include <iostream>
#include <cstdio>
using namespace std;
typedef long long i64;
const int MOD = 1000000007;
int n, k;
char s[505];
i64 f[505][505][6];
// f[0]: **
// f[1]: (..)
// f[2]: (..)**
// f[3]: (..)..(..) 包含 1
// f[4]: **(..)
// f[5]: **(..)** 包含 0
int main(void)
{
scanf("%d%d%s", &n, &k, s + 1);
for (int i = 1; i <= n; ++i) f[i][i-1][0] = 1;
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
if (len <= k) f[i][j][0] = f[i][j-1][0] && (s[j] == '*' || s[j] == '?');
if (len >= 2) {
if ((s[i] == '(' || s[i] == '?') && (s[j] == ')' || s[j] == '?'))
f[i][j][1] = (f[i+1][j-1][0] + f[i+1][j-1][2] + f[i+1][j-1][3] + f[i+1][j-1][4]) % MOD;
for (int k = i; k < j; ++k) {
f[i][j][2] = (f[i][j][2] + f[i][k][3] * f[k+1][j][0]) % MOD;
f[i][j][3] = (f[i][j][3] + (f[i][k][2]+f[i][k][3]) * f[k+1][j][1]) % MOD;
f[i][j][4] = (f[i][j][4] + (f[i][k][4]+f[i][k][5]) * f[k+1][j][1]) % MOD;
f[i][j][5] = (f[i][j][5] + f[i][k][4] * f[k+1][j][0]) % MOD;
}
}
f[i][j][5] = (f[i][j][5] + f[i][j][0]) % MOD;
f[i][j][3] = (f[i][j][3] + f[i][j][1]) % MOD;
}
printf("%lld\n", f[1][n][3]);
return 0;
}
高维 DP
高维 DP。
[SDOI2010] 地精部落
一座长度为 的山脉可分为从左到右的 段,每段有一个独一无二的高度 ,其中 是 到 之间的正整数。这 段山脉每段都必须是一个山峰或者山谷,问这样的山脉有多少种。答案对 取模。,。
设 代表考虑前 个数,第一个数是山顶,高度为 的方案数。当 与 不相邻的时候,两者位置可以交换,方案数是 ;当相邻的时候,由于 是山峰, 是山谷,这时我们将第 个数翻转成 ,这时候它就成了山峰,方案数就是 。
由于我们假设了第一个数是山顶,所以还有第一个数是山谷的答案,答案要乘 。
查看代码
#include <iostream>
#include <cstdio>
using namespace std;
int n, p;
int f[2][4205];
int main(void) {
scanf("%d%d", &n, &p);
f[0][2] = 1;
for (int i = 3; i <= n; ++i)
for (int j = 2; j <= i; ++j)
f[i & 1][j] = (f[i & 1][j - 1] + f[(i - 1) & 1][i - j + 1]) % p;
uint ans = 0;
for (int i = 1; i <= n; ++i)
ans = (ans + f[n & 1][i] * 2) % p;
printf("%u\n", ans);
return 0;
}
线性 DP 的综合应用
这里的题难度会稍大。
[UVA10559] Blocks
预处理颜色段,然后定义 代表消除 ,且 右边还有 个与 颜色相同的木块。那么要么是消 ,要么是在 中有与 相同的颜色,消完中间段之后将两段颜色合并再进行消除。
查看代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, a[205];
int tot, len[205], color[205];
int f[205][205][205];
int dfs(int l, int r, int k) {
if (f[l][r][k]) return f[l][r][k];
if (l == r) return f[l][r][k] = (len[r] + k) * (len[r] + k); // 只有一段,只能消除
f[l][r][k] = dfs(l, r - 1, 0) + (len[r] + k) * (len[r] + k); // 先消除 [l, r - 1]
for (int i = l; i < r - 1; ++i) // 在 [l, r - 2] 中寻找与 r 相同的颜色
// 先消除 [i + 1, r - 1]
// 然后 [l, i] 就会与 [r, r + len[r] + k] 合并,可以直接消除
if (color[i] == color[r]) f[l][r][k] = max(f[l][r][k], dfs(l, i, len[r] + k) + dfs(i + 1, r - 1, 0));
return f[l][r][k];
}
int main(void) {
int T, kase = 0;
scanf("%d", &T);
while (T--) {
memset(f, 0, sizeof(f));
tot = 0;
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
scanf("%d", a + i);
if (a[i] == a[i - 1]) ++len[tot];
else ++tot, len[tot] = 1, color[tot] = a[i];
}
n = tot;
printf("Case %d: %d\n", ++kase, dfs(1, n, 0));
}
return 0;
}
[THUSC2016] 成绩单
期末考试结束了,班主任 L 老师要将成绩单分发到每位同学手中。L 老师共有 份成绩单,按照编号从 到 的顺序叠放在桌子上,其中编号为 的的成绩单分数为 。
成绩单是按照批次发放的。发放成绩单时,L 老师会从当前的一叠成绩单中抽取连续的一段,让这些同学来领取自己的成绩单。当这批同学领取完毕后,L 老师再从剩余的成绩单中抽取连续的一段,供下一批同学领取。经过若干批次的领取后,成绩单将被全部发放到同学手中。对于一个分发成绩单的方案,我们定义其代价为:
其中 是分发的批次数,对于第 披分发的成绩单, 是最高分数, 是最低分数, 和 是给定的评估参数。现在,请你帮助 L 老师找到代价最小的分发成绩单的方案,并将这个最小的代价告诉 L 老师。当然,分发成绩单的批次数 是你决定的。
。
设 代表区间 全部取走的答案。初始时 ,答案是 。现在考虑转移,我们设 代表区间 中所有成绩单(可以有发走的)中的最大值为 ,最小值为 时,发放的最小代价,那么:
显然在 DP 之前需要先离散化,现在考虑 如何求解。
可以是 和 合并得来的,也可以是从 转移过来的。
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define rep(i, l, r) for (int i = l; i <= r; ++i)
using namespace std;
inline int F(int x) { return x * x; }
int n, a, b;
int w[55], l[55];
int f[55][55][55][55], g[55][55];
int main(void) {
scanf("%d%d%d", &n, &a, &b);
for (int i = 1; i <= n; ++i)
scanf("%d", w + i), l[i] = w[i];
sort(l + 1, l + n + 1);
int m = unique(l + 1, l + n + 1) - (l + 1);
for (int i = 1; i <= n; ++i)
w[i] = lower_bound(l + 1, l + m + 1, w[i]) - l;
memset(f, 0x3f, sizeof(f));
memset(g, 0x3f, sizeof(g));
for (int i = 1; i <= n; ++i)
f[i][i][w[i]][w[i]] = 0, g[i][i] = a;
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
rep(x, 1, m) rep(y, x, m) {
f[i][j][min(x, w[j])][max(y, w[j])] = min(f[i][j][min(x, w[j])][max(y, w[j])], f[i][j - 1][x][y]);
for (int k = i; k < j; ++k)
f[i][j][x][y] = min(f[i][j][x][y], f[i][k][x][y] + g[k + 1][j]);
}
rep(x, 1, m) rep(y, x, m)
g[i][j] = min(g[i][j], f[i][j][x][y] + a + b * F(l[y] - l[x]));
}
printf("%d\n", g[1][n]);
return 0;
}
[POI2015] Myjnie
有 家洗车店从左往右排成一排,每家店都有一个正整数价格 。有 个人要来消费,第 个人会驶过第 个开始一直到第 个洗车店,且会选择这些店中最便宜的一个进行一次消费。但是如果这个最便宜的价格大于 ,那么这个人就不洗车了。请给每家店指定一个价格,使得所有人花的钱的总和最大。
先将 离散化,然后考虑 代表考虑 的洗车间,它们当中的最小消费额大于等于 。初始时 转移的时候很简单:
其中 代表满足 的人的个数,这个可以在枚举 的时候预处理。需要输出方案,在转移的时候需要记录当前这个状态的最小消费额和分割点,然后 dfs 输出,分割点的答案记录为最小消费额。
查看代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int n, m;
int a[4005], b[4005], c[4005], l[4005];
int f[55][55][4005]; // 最小值大于等于 k
int cnt[55][55];
pair<int, int> p[55][55][4005];
int ans[55];
void dfs(int i, int j, int k) {
if (i > j) return;
pair<int, int> it = p[i][j][k];
ans[it.second] = l[it.first];
dfs(i, it.second - 1, it.first); dfs(it.second + 1, j, it.first);
}
int main(void) {
scanf("%d%d", &n, &m);
for (int i = 1; i <= m; ++i)
scanf("%d%d%d", a + i, b + i, c + i), l[i] = c[i];
sort(l + 1, l + m + 1);
int t = unique(l + 1, l + m + 1) - (l + 1);
for (int i = 1; i <= m; ++i)
c[i] = lower_bound(l + 1, l + t + 1, c[i]) - l;
for (int op = t; op >= 1; --op) {
for (int k = 1; k <= m; ++k) if (c[k] == op) {
for (int i = 1; i <= a[k]; ++i)
for (int j = b[k]; j <= n; ++j)
++cnt[i][j];
}
for (int len = 1; len <= n; ++len)
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
f[i][j][op] = f[i][j][op + 1];
p[i][j][op] = p[i][j][op + 1];
for (int k = i; k <= j; ++k) {
int v = f[i][k - 1][op] + f[k + 1][j][op] + (cnt[i][j] - cnt[i][k - 1] - cnt[k + 1][j]) * l[op];
if (v > f[i][j][op]) {
f[i][j][op] = v;
p[i][j][op] = {op, k};
}
}
if (p[i][j][op].first == 0) p[i][j][op] = {op, i};
}
}
printf("%d\n", f[1][n][1]);
dfs(1, n, 1);
for (int i = 1; i <= n; ++i) printf("%d ", ans[i]);
putchar('\n');
return 0;
}
小结
区间 DP 和高维 DP 是线性结构上比较难的一类 DP,但只要多写题,也不是什么难事。