[线性dp] 数字三角形(模板+数字三角形模型)

0. 前言

线性 dp 指的是状态转移方程有明显的线性关系，其可以是一维线性关系也可以是二维线性关系。

也可从建立的 dp 数组中观察，数组中 dp 值是前后依次更新的，称为具有线性关系。

线性 dp 问题占 dp 问题的比重很大，思路比较好想。

1. 数字三角形+模板题

898. 数字三角形

重点： 线性 dp、边界初始化

思路：

• 将数字三角形按照行、斜列顺序从 1 开始编号
• 状态定义：
  • f[i][j] 表示从起点走到 [i,j] 点的所有路径数字的最大值
• 状态转移：
  • 来自左上方：f[i-1][j-1] + a[i][j]
  • 来自右上方：f[i-1][j] + a[i][j]
  • 则状态转移方程为：f[i][j] = max(f[i-1][j-1], f[i-1][j]) + a[i][j]
• 边界处理及初始化
  • 由于出现 i-1，则存储下标从 1 开始避免边界情况
  • 由于是取 max，则先将 f 数组全部置为 -inf。这里 memset 初始化负无穷的时候建议使用 0xc0c0c0c0=-1061109568 ，0x3f3f3f3f = 1061109567
  • f[1][1] = a[1][1]
  • 采用循环初始化的时候需要注意：在数字三角形边界进行状态转移的时候会跑到边界外进行状态转移。所以初始化 -inf 的时候需要多初始化一圈，把边界外的一圈也包裹起来。这个是 for 进行 f 数组部分初始化务必需要记住的一点。

自顶向下代码：

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 510, INF = 1e9;

int n;
int a[N][N];
int f[N][N];

int main() {
    cin >> n;
    
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= i; ++j)
            cin >> a[i][j];
            
    // memset(f, 0xc0, sizeof f);
    
    for (int i = 0; i <= n; ++i) 
        for (int j = 0; j <= i + 1; ++j)	// 初始化注意
            f[i][j] = -INF;

    f[1][1] = a[1][1];
    
    for (int i = 2; i <= n; ++i)
        for (int j = 1; j <= i; ++j)    
            f[i][j] = max(f[i - 1][j - 1], f[i -1][j]) + a[i][j];
    
    int res = -INF;
    for (int i = 1; i <= n; ++i) res = max(res, f[n][i]);
    
    cout << res << endl;
    
    return 0;
}

以上思路为自顶向下求解，很正常的思路，但需要考虑边界情况。且需要遍历最后一行得到最大值。

也可以类似考虑自底向上求解，不需要考虑边界及最大值。因为其保证了当前数字一定存在左下数字和右下数字，且不需要初始化，因为直接是针对三角形内的现有数字取 max 的，不存在边界问题。输出答案的时候直接就是顶部 f[1][1]，是非常优秀的写法。

自底向上代码：

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 510, INF = 1e9;

int n;
int a[N][N];
int f[N][N];

int main() {
    cin >> n;
    
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= i; ++j)
            cin >> a[i][j];
    
    for (int i = n; i >= 1; --i)
        for (int j = i; j >= 1; --j)    
            f[i][j] = max(f[i + 1][j + 1], f[i + 1][j]) + a[i][j];
    
    cout << f[1][1] << endl;
    
    return 0;
}