文章目录
0. 前言
线性 dp
指的是状态转移方程有明显的线性关系,其可以是一维线性关系也可以是二维线性关系。
也可从建立的 dp
数组中观察,数组中 dp
值是前后依次更新的,称为具有线性关系。
线性 dp
问题占 dp
问题的比重很大,思路比较好想。
1. 数字三角形+模板题
重点: 线性 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;
}