第八周课后作业,LeetCode上的题目选一道写一下
题目链接:https://leetcode-cn.com/problems/triangle/description/
第120题 三角形最小路径和
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[2],
[3,4],
[6,5,7],
[4,1,8,3]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
分析:
这是一道简单的动态规划,唯一的难点是题目里给的空间复杂度的要求。当选择下面一层中的数字时,我们只能选择相邻的数字。对于一个data[i][j],和它相邻的数字就是data[i+1][j]和data[i+1][j+1]。假如用minimus[i][j]来表示从第i行第j列处的数字开始往下到最后一层的最小路径和,那么有minimus[i][j]=data[i][j]+min(minimums[i+1][j]+minimums[i+1][j+1])
然而上述描述中需要一个O(n^2)的额外空间。
不妨反过来想一下,先求解子问题,然后再解决父问题。即,从下往上求解最小路径和。
当求解minimum[i][j]时,会用到minimum[i+1][j]和minimum[i+1][j+1],但是当求解完所有minimum[i]之后minimum[i+1]就没有用处了,可以复用同一个空间来存储minimum的值。存储最后一行的每个数字的最小路径和需要n个空间,之后存储倒数第二行时,只需要前面的n-1个空间……以此类推,第一行只需要一个空间来存储最小路径和,这正是求解的结果。
代码如下
class Solution:
def minimumTotal(self, triangle):
"""
:type triangle: List[List[int]]
:rtype: int
"""
i = len(triangle)
minlength = triangle[i-1]
i = i - 2
while i >= 0:
for j in range(0, len(triangle[i])):
minlength[j] = min(minlength[j], minlength[j+1]) + triangle[i][j]
i = i - 1
return minlength[0]