Leetcode 120 三角形最小路径和

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

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

 

输入:triangle = [[-10]]

输出:-10

分析:

  一开始就能想到回溯法,遍历每一个路径,但是我们要想到,能用自顶向上的回溯法,就要考虑自底向上的动态规划

  因为对于3来说,要找到6和5的答案, 对于4来说,要找到5和7的答案,回溯法显然会冗余

因为每一层只与下一层有关系,但是一层实现不了,所以两层

def minimumTotal(triangle) -> int:
max_len = len(triangle[-1])
dp = [[0 for i in range(max_len)] for j in range(2)]
pre = -1
for i in range(len(triangle)-1, -1, -1):
idx = i%2
pre = idx
idx2 = (idx+1)%2
if i == len(triangle)-1:
for j in range(max_len):
dp[idx][j] = triangle[i][j]
else:
for j in range(len(triangle[i])):
dp[idx][j] = triangle[i][j] +min(dp[idx2][j], dp[idx2][j+1])
return dp[pre][0]
posted @ 2022-06-15 01:29  王毅2016  阅读(17)  评论(0编辑  收藏  举报