题目链接
题目描述
给定一个三角形 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 为三角形的总行数)来解决这个问题吗?
解题思路
只用一维dp的方法:
不考虑边界的情况下,对于新一行的第i个元素来说,dp[i]和dp[i-1]存储了上一行的最小路径,该元素可以从这两个里取较小的,在其进行完取舍后又可以把自己的最小路径存储在dp[i]上,于是dp[i-1]就需要提前用一个变量存储即可
代码
Python
class Solution:
def minimumTotal(self, triangle: list[list[int]]) -> int:
depth = len(triangle)
if depth == 1:
return triangle[0][0]
dp = [201 for _ in range(depth)]
dp[0] = triangle[0][0]
for i in range(1, depth):
tmp = dp[0]
for j in range(0, i + 1):
if j == 0 or j == i:
dp[j] = tmp + triangle[i][j]
else:
left = tmp
tmp = dp[j]
dp[j] = min(left, dp[j]) + triangle[i][j]
return min(dp)