【线性DP】数字三角形

题目链接

原题链接

题目描述

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

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

输入格式

第一行包含整数 n，表示数字三角形的层数。

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

输出格式

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

数据范围

1≤n≤500,
−10000≤三角形中的整数≤10000

输入样例：

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

输出样例：

30

难度：简单

时/空限制：1s / 64MB

来源：模板题, usaco training 1.6

算法标签

动态规划、线性DP

思路

代码

点击查看代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

const int N = 510;
const int INF = 1e9;

int n;
int a[N][N] , 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];

	//初始化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]+a[i][j] , f[i-1][j]+a[i][j] );

	int ans = -INF;
	for(int i = 1;i <= n;i ++)
		ans = max(ans , f[n][i]);

	cout << ans << endl;

	return 0;
}