動態規劃--三角形最小路徑和

星空28發表於2024-10-04

問題描述

給定一個三角形 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])

相關文章