leetcode_120_三角形最小路径和

三角形最小路径和

描述

中等

给定一个三角形，找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。

相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。

例如，给定三角形：

[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]
自顶向下的最小路径和为 11（即，2 + 3 + 5 + 1 = 11）。

说明：

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

解题

将数据左对齐看

[
  [2],
  [3,4],
  [6,5,7],
  [4,1,8,3]
]

• 当为最左端节点时，其路径只能从上到下
  $$f(x,y)=f(x-1,y)+t(x,y)$$

• 当为最右端节点时，其路径只能从左上过来
  $$f(x,y)=f(x-1,y-1)+t(x,y)$$

• 每一个节点路径只能从左上或上方过来
  $$f(x,y)=min(f(x-1,y-1),f(x,y-1))+t(x,y)$$

如果申请dp[n][n]，空间复杂度就直接不满足了

所以申请dp[n]，然后每次在dp[n]的空间内进行更新

class Solution:
    def minimumTotal(self, triangle: List[List[int]]) -> int:
        if not triangle:
            return 0

        n = len(triangle)
        dp = [0 for i in range(n)]
        dp[0] = triangle[0][0]
        

        pre = 0 # 用于记录“左上方的点”的路径和
        cur = 0 # 用于记录“上方的点”的路径和
        for i in range(1, n):
            for j in range(i+1):
                cur = dp[j]
                if j == 0:
                  	# 最左端的点
                    dp[j] = cur + triangle[i][j]
                elif j == i:
                    # 最右端的点
                    dp[j] = pre + triangle[i][j]
                else:
                    dp[j] = min(cur, pre) + triangle[i][j]
                pre = cur
        
        return min(dp)