AcWing 898 数字三角形 题解（动态规划—DP—数塔）

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 1e4 + 10, INF = 1e9;

int a[N][N];
int h[N][N];//记录到达每个点的最大路径数字之和 
int n;

int main()
{
	cin>>n;
	for(int i = 1; i <= n; i ++ ){
		for(int j = 1; j <= i; j ++ ){
			cin>>a[i][j];//从1开始读入数塔的数，防止有边界问题 
		}
	}
	
	for(int i = 0; i <= n; i ++ ){//考虑到数塔最右侧的边再往右的行数为0的一行也会被考虑，所以也需要初始化 
		for(int j = 0; j <= i + 1; j ++ ){//考虑到数塔最右侧的边再往右的行数为0的一行和最左侧再往左的列数为0的一列也会被考虑，所以每行的第j = i + 1和j = 0个数也需要初始化 
			h[i][j] = -INF; 
		}
	}
	
	h[1][1] = a[1][1];//初始化第一行起点
	
	for(int i = 2; i <= n; i ++ ){//第一行为起点，从第二行开始更新
		for(int j = 1; j <= i ; j ++ ){
			h[i][j] = max(h[i - 1][j - 1] + a[i][j], h[i - 1][j] + a[i][j]);//状态转移方程，分别考虑一个点的左上邻接点和右上邻接点 
		}
	}
	
	int res = -INF;
	for(int i = 1; i <= n; i ++ ){
		res = max(res, h[n][i]);//遍历数塔最底层，找出最大路径数字之和 
	} 
	cout<<res<<endl;
	return 0;
}