链接:
题目:
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[ [2], [3,4], [6,5,7], [4,1,8,3] ]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
我的解法:在triangle矩阵里用min(triangle[i-1][j-1], triangle[i-1][j]) + triangle[i][j] 赋值给triangle[i][j],最后最后一行的最小值即为所求。
class Solution(object):
def minimumTotal(self, triangle):
if not triangle:
return 0
if len(triangle) == 1:
return triangle[0][0]
for i in range(1, len(triangle)):
for j in range(len(triangle[i])):
if 0 < j < len(triangle[i])-1:
triangle[i][j] = min(triangle[i-1][j-1], triangle[i-1][j]) + triangle[i][j]
elif j == 0:
triangle[i][j] = triangle[i-1][j] + triangle[i][j]
else:
triangle[i][j] = triangle[i-1][-1] + triangle[i][j]
return min(triangle[-1])
一个意思,写了另一种写法:
class Solution(object):
def minimumTotal(self, triangle):
if not triangle:
return 0
if len(triangle) == 1:
return triangle[0][0]
for i in range(1, len(triangle)):
for j in range(i+1):
left = 1e10 if j == 0 else triangle[i-1][j-1]
right = 1e10 if j == i else triangle[i-1][j]
triangle[i][j] += min(left, right)
return min(triangle[-1])