【ACWing】898. 数字三角形

题目地址：

https://www.acwing.com/problem/content/900/

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

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

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

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

数据范围：
$1\le n\le 500$
$-10000\le x\le 10000$
$x$为三角形内数字

思路是动态规划。设$f[i][j]$是走到$(i,j)$这个位置的所有路径中，路径和的最大值。那么这个路径可能是从正上方走过来的，也可能是从左上方走过来的。所以有 f [ i ] [ j ] = max ⁡ { f [ i − 1 ] [ j − 1 ] , f [ i − 1 ] [ j ] } + a [ i ] [ j ] f[i][j]=\max\{f[i-1][j-1],f[i-1][j]\}+a[i][j] f[i][j]=max{f[i−1][j−1],f[i−1][j]}+a[i][j]代码如下：

#include <iostream>
using namespace std;

const int N = 510, INF = 1e9;
int a[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];

  for (int i = 2; i <= n; i++)
    for (int j = 1; j <= i; j++) {
      int t = -INF;
      if (j < i) t = max(t, a[i - 1][j]);
      if (j > 1) t = max(t, a[i - 1][j - 1]);

      a[i][j] += t;
    }

  int res = a[n][1];
  for (int i = 2; i <= n; i++) res = max(res, a[n][i]);

  cout << res << endl;
}

时间复杂度$O(n^2)$，空间$O(1)$。