120.三角形最小路径和
给定一个三角形 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)。
动态规划:
dp[i,j] = min(dp[i+1][j],dp[i+1][j+1]) + triangle[i][j]
自底向上遍历,找到每个元素相对应下层两个相邻元素的最小值,并进行累加,最后直接返回数组的第一个元素,即为最小路径和
解法一:(直接复用原数组,不再开辟新的空间)
自底向上遍历 Python3版本
class Solution:
def minimumTotal(self, triangle: List[List[int]]) -> int:
if not triangle: # 先进行判空操作
return
# 从二维数组的倒数第二行开始往上遍历,因为倒数第一行不再有下一层
for i in range(len(triangle) - 2, -1, -1): # 行遍历
for j in range(len(triangle[i])): # 列遍历
triangle[i][j] += min(triangle[i+1][j],triangle[i+1][j+1]) # 状态转移方程
return triangle[0][0] # 返回三角形顶部元素,即为最小路径和
把range改成xrange就成了Python版了,Python版的运行速度较Python3版要快不少
解法二:(用一个一维数组对结果进行保存)
这个是Python版本,在Python3中运行会报错,因为Python3中没有xrange
这个运行速度也略快一点
class Solution(object):
def minimumTotal(self, triangle):
"""
:type triangle: List[List[int]]
:rtype: int
"""
if not triangle:
return
res = triangle[-1] # 将triangle的最后一行赋值给res
# 遍历依然是从倒数第二行开始,因为倒数第一行的值已经保存且无下一层
for i in xrange(len(triangle) - 2, -1, -1): # 行遍历
for j in xrange(len(triangle[i])): # 列遍历
res[j] = min(res[j],res[j+1]) + triangle[i][j]
return res[0]
range与xrange的区别如下:
1、range和xrange都是在循环中使用,输出的结果是一样的
2、range返回的是一个list对象,而xrange返回的是一个生成器对象(xrange object)
3、xrange不会直接生成一个list,而是每次调用返回其中的一个值,内存空间使用极少,因而性能非常好
遗憾的是在Python3中使用xrange会抛出异常