AcWing 898. 数字三角形(线性dp)

题目链接： 点击查看

题目描述：

给定一个如下图所示的数字三角形，从顶部出发，在每一结点可以选择移动至其左下方的结点或移动至其右下方的结点，一直走到底层，要求找出一条路径，使路径上的数字的和最大。

输入输出格式：

输入

第一行包含整数 n，表示数字三角形的层数。
接下来 n 行，每行包含若干整数，其中第 i 行表示数字三角形第 i 层包含的整数。

输出

输出一个整数，表示最大的路径数字和。

输入输出样例：

输入

5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5

输出

30

题目分析：

dp问题的解法可以从以下两方面来考虑：状态表示和状态计算。数字三角形是经典的线性dp问题，其状态表达f[i, j]含义是表示从起点到(i, j)点所有路径的集合，其属性是所有路径中最大的路径数字和。状态计算(转移方程)为 f[i, j] = max(f[i - 1, j - 1] + a[i, j], f[i - 1, j] + a[i, j]); 含义如下

比如我们选取点(4, 2) 值为7的点，最终到达该点的路径一共来自两个方向：左上和右上，我们用f[i, j]存储最优解，每一个点的最优解都可以用上一个点的最优解来得到。如果从左上方向来到该点，状态转移方程为f[i,j] = f[i - 1, j - 1] + a[i, j]; 如果从右上方向来到该点, 其状态转移方程为f[i, j] = f[i - 1, j] + a[i, j]。要注意的是，并不是每个点都有左上或右上的点，要注意对边界情况进行处理。详见如下代码。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
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 = 0; i <= n; i ++ ) {
		for (int j = 0; j <= i + 1; j ++ ) {//初始化到i + 1是因为有负数存在的情况，要将两边也初始化为负无穷 
			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] + a[i][j], 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;
	return 0;
	 
}