算法题 动态规划-数字三角形（Python）

题目

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

        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

代码

n = int(input())
nums = []
for _ in range(n):
    nums.append(list(map(int, input().split())))
    
dp = [0] * (n)
dp[0] = nums[0][0]
for i in range(1, n):
    for j in range(len(nums[i])-1, -1, -1):
        if j == len(nums[i]) - 1: dp[j] = dp[j-1]
        elif j != 0: dp[j] = max(dp[j], dp[j-1])
        dp[j] += nums[i][j]
print(max(dp))