【python】数字三角形——动态规划问题

题目描述

上图给出了一个数字三角形，从顶部出发，在每一结点可以选择移动至其左下方的结点或移动至其右下方的结点，一直走到底层，要求找出一条路径，使路径上的数字的和最大。
路径上的每一步只能从一个数走到下一层和它最近的左边的那个数或者右边的那个数。此外，向左下走的次数与向右下走的次数相差不能超过 1。

输入输出样例

示例

输入

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

输出

27

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

思路分析

典型的一类动态规划问题求最大值，多了一个限制条件：向左下走的次数与向右下走的次数相差不能超过 1。通过网上的思路和自我进行几个例子的演示，看了发现其实最终答案与我们输入的N的奇偶性有关：

当n为奇数，最终答案为中间的数；当n为偶数，最终答案为中间两个值中的最大值。

代码实现

• 法一

import os
import sys

n=int(input())
num=[]
for _ in range(n):
    num.append(list(map(int,input().split()))) 
    ## 使用map函数将输入的字符串中的数字转为int类型, 最终使用list方法将其转换为一个列表
    
for i in range(1,n):
    for j in range(i+1):
        if j == 0:    #最左边一列
            num[i][j] += num[i-1][j]
        elif j==i:    #最右边一列
            num[i][j]