【leetcode】#陣列【Python】120. Triangle 三角形最小路徑和
阿新 • • 發佈:2018-12-21
連結:
題目:
給定一個三角形,找出自頂向下的最小路徑和。每一步只能移動到下一行中相鄰的結點上。
例如,給定三角形:
[ [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])