Skip to content

区间动态规划

石子合并

每次只能合并相邻的两堆,合并的代价为这两堆石子的质量之和,合并后与这两堆石子相邻的石子将和新堆相邻,合并时由于选择的顺序不同,合并的总代价也不相同。

cpp
4
1 3 5 2 -> 1 3 7 -> 4 7 -> 11
         7        4     11    (22)

显然不能用贪心来做。

  • 状态表示:f[i][j] 去掉所有其他石子堆,只研究 [i,j] 区间内的石子堆,此时所有将第 i 堆石子到第 j 堆石子合并成一堆石子的合并方式(代价的最小值),而所求的最小值就是 f[1][n]
  • 状态计算:合并一定存在最后一次合并,因此以最后一次合并所在的分界线位置来分类
    • 分界线有 1 ~ n - 1n1 种,而对于 f[i][j] 只有 i ~ j - 1 一共 ji
    • 于是可以分为 ji 类(或者可以这么分类:按照最后一次合并时左边已经合并了 [i,j] 区间内的多少堆来分)
    • 一定要注意:f[i][j] 表示去除了其他的石子堆后的结果,因此任何一种合并方式最后一步的代价都是相同的,这里可以用前缀和来求解 a[i]+...+a[j] = s[j] - s[i-1];
    • 而剩下的 ji 类合并则是 f[i][k] + f[k+1][j],其中 k=i,,j1
  • 时间复杂度为 O(n3)

AC282. 石子合并

image-20231202053000505

如上图所示,这是区间 DP 相较于线性 DP 的一大不同,打表方式这里是斜对角线依次打表,先算出 4, 7, 6,再算出 12, 15,最后算出答案 22。对于这种情况下的下三角矩阵,我们的循环代码为:

cpp
// normal
for (int i = 1; i <= n; i ++) {
    for (int j = 1; j <= i; j ++) {
        cin >> f[i][j];
    }
}

// diagonal first
for (int t = 0; t < n; t ++) {
    for (int i = 1; i <= n; i ++) {
        cin >> f[i + t][i];
    }
}

它是先算:

cpp
f[1][1] f[2][2] f[3][3]
f[2][1] f[3][2]
f[3][1]

体现的是 f[i][j]j-i 的大小。因此也可以说这是以 长度 作为第一重循环变量,这也是区间 DP 的循环方式。

核心代码:

cpp
// ...
    for (int len = 2; len <= n; len ++) {
        for (int i = 1; i + len - 1 <= n; i ++) {
            int l = i;
            int r = i + len - 1;
            
            f[l][r] = 1e9;

            for (int k = l; k < r; k ++) {
                f[l][r] = min(f[l][r], f[l][k] + f[k + 1][r] + s[r] - s[l - 1]);
            }
        }
    }
  • 长度为 1 的 f[i][i] 自然是 0,从长度为 2 开始计算
  • 由于取的是最小值,因此先给 f[l][r] 一个较大的数,之后会舍掉

递归写法(记忆化搜索)

Released under the MIT License.