1. 程式人生 > >【leetcode】#陣列【Python】120. Triangle 三角形最小路徑和

【leetcode】#陣列【Python】120. Triangle 三角形最小路徑和

連結:

題目:

給定一個三角形,找出自頂向下的最小路徑和。每一步只能移動到下一行中相鄰的結點上。

例如,給定三角形:

[
     [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])