120. 三角形最小路径和（python3）

题目：https://leetcode-cn.com/problems/triangle/

给定一个三角形 triangle ，找出自顶向下的最小路径和。

每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说，如果正位于当前行的下标 i ，那么下一步可以移动到下一行的下标 i 或 i + 1 。

 

示例 1：

输入：triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出：11
解释：如下面简图所示：
   2
  3 4
 6 5 7
4 1 8 3
自顶向下的最小路径和为 11（即，2 + 3 + 5 + 1 = 11）。
示例 2：

输入：triangle = [[-10]]
输出：-10
 

提示：

1 <= triangle.length <= 200
triangle[0].length == 1
triangle[i].length == triangle[i - 1].length + 1
-104 <= triangle[i][j] <= 104
 

进阶：

你可以只使用 O(n) 的额外空间（n 为三角形的总行数）来解决这个问题吗？

 

解题思路：

动态规划问题，不太了解解题思路的可先参考：https://blog.csdn.net/anningne/article/details/112284336

转移方程为dp[i][j] = min(dp[i-1][j], dp[i-1][j-1]) + triangle[i][j]，初始状态为dp[0][0]，需注意j==0的情况j-1越界的情况及i==j的情况（每行最右侧数=左上方数+上方数，上方数初始化一定要置为最大数，否则在判断最小数时会出错），动态规划的转移方程一般以最后一步得出，此处尤其注意不要将转移方程写成dp[i] = dp[i-1]+min(triangle[i][k]],triangle[i][k+1]),这种情况相当于每一步都是找的局部最优而非全局最优。倒推时要以落的最后一步位置往前推，最后一行的每一个位置都是由上一行两条路径决定。

 

代码：

import sys

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:
        m = len(triangle)

        key = sys.maxsize
        dp = [[key] * m for _ in range(m)]
        
        if(m == 0):
            return 0

        dp[0][0] = triangle[0][0]

        if(m == 1):
            return dp[0][0]

        for i in range(1, m):
            for j in range(0, i + 1):
                if(j == 0):
                    dp[i][j] = dp[i - 1][j] + triangle[i][j]
                else:
                    dp[i][j] = min(dp[i - 1][j], dp[i - 1][j - 1]) + triangle[i][j]
           
        minvalue = key
        for i in range(m):
            minvalue = dp[m - 1][i] if dp[m - 1][i] < minvalue else minvalue     
                   
        return minvalue

 

测试：

if __name__ == "__main__":
    triangle = [[-1],[2,3],[1,-1,-3]]
    res = Solution().minimumTotal(triangle)
    print(res)