题目描述
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
解题方法
考虑将问题转化为子问题:从三角形顶层到底层顺序,已知从顶层节点走到上一层中各个节点的最短路径,求从顶层到当前层各个节点的最短路径可转换为上一层到当前层的最短路径求解。
状态定义
1. 定义dp形状和三角形相同形状,dp[i][j]表示从三角形顶层到三角形i层j节点时经历的最短路径。
2. 初始化dp[0][0] = 三角形顶点的值
状态转移
dp[i][j] = triangle[i][j] + min(dp[i-1][父节点])
父节点表示i层j节点在上一层中相邻的节点id。
class Solution:
def minimumTotal(self, triangle: List[List[int]]) -> int:
def parent_id(i,j):
return [max(0,j-1),min(j,i-1)]
dp = triangle
for i in range(1,len(triangle)):
for j in range(len(triangle[i])):
p_id = parent_id(i,j)
dp[i][j] = triangle[i][j] + min(dp[i-1][p_id[0]],dp[i-1][p_id[1]])
return min(dp[-1])