leetcode 120 三角形最小路径和
把三角形看成一个上三角或者下三角(等腰三角形),然后设置状态转移公式,当没到每行的最后一个的时候,以及每行的第一个,都可以有两种方法到达当前位置,选择最小的路径和。
class Solution: def minimumTotal(self, triangle: List[List[int]]) -> int: dp=[[0 for _ in range(len(triangle[-1]))] for i in range(len(triangle))] dp[0][0]=triangle[0][0] for i in range(1,len(triangle)): dp[i][0]=dp[i-1][0]+triangle[i][0] #dp[1][1]=triangle[0][0]+triangle[1][1] for i in range(1,len(triangle)): for j in range(1,i+1): if j==i: dp[i][j]=dp[i-1][j-1]+triangle[i][j] else: dp[i][j]=min(dp[i-1][j-1]+triangle[i][j],dp[i-1][j]+triangle[i][j]) print(dp) return min(dp[len(triangle)-1])
空间优化,通过递减的方式实现仅使用一维数组的空间。
class Solution: def minimumTotal(self, triangle: List[List[int]]) -> int: dp=[0 for _ in range(len(triangle[-1]))] dp[0]=triangle[0][0] for i in range(1,len(triangle)): for j in range(i,-1,-1): if j==i: dp[j]=dp[j-1]+triangle[i][j] elif j==0: dp[0]=dp[0]+triangle[i][j] else: dp[j]=min(dp[j-1]+triangle[i][j],dp[j]+triangle[i][j]) return min(dp)