問題描述
給定一個三角形 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)。
示例 2:
輸入: triangle = [[-10]]
輸出: -10
程式碼
class Solution(object):
def minimumTotal(self, triangle):
"""
:type triangle: List[List[int]]
:rtype: int
"""
min_sum = []
for i in range(len(triangle)):
sum_i_min = []
if i == 0:
sum_i_min.append(triangle[i][0])
else:
for j in range(len(triangle[i])):
if j == 0:
print(sum_i_min)
sum_i_min.append(triangle[i][j] + min_sum[i-1][j])
elif j == len(triangle[i]) - 1:
sum_i_min.append(triangle[i][j] + min_sum[i-1][j-1])
else:
sum_i_min.append(min(triangle[i][j] + min_sum[i-1][j-1], triangle[i][j] + min_sum[i-1][j]))
min_sum.append(sum_i_min)
return min(min_sum[-1])