LeetCode 120. 三角形最小路径和

Posted 2020-12-17 机器狗mo

给定一个三角形，找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。?

例如，给定三角形：
[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]
自顶向下的最小路径和为?11（即，2?+?3?+?5?+?1?= 11）。?

说明：
如果你可以只使用 O(n)?的额外空间（n 为三角形的总行数）来解决这个问题，那么你的算法会很加分。

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:
        ans = [0]
        for i in range(len(triangle)):
            new_ans = []
            for j in range(len(triangle[i])):
                if j == 0:
                    new_ans.append(triangle[i][j]+ans[j])
                elif j==len(triangle[i])-1:
                    new_ans.append(triangle[i][j]+ans[j-1])
                else:
                    new_ans.append(triangle[i][j]+min(ans[j-1],ans[j]))
            ans = new_ans
            #print(‘new_ans‘,new_ans)
        return min(ans)