給定一個方形整數陣列 A，我們想要得到通過 A 的下降路徑的最小和。

下降路徑可以從第一行中的任何元素開始，並從每一行中選擇一個元素。在下一行選擇的元素和當前行所選元素最多相隔一列。

示例：

輸入：[[1,2,3],[4,5,6],[7,8,9]]
輸出：12
解釋：
可能的下降路徑有：

• [1,4,7], [1,4,8], [1,5,7], [1,5,8], [1,5,9]
• [2,4,7], [2,4,8], [2,5,7], [2,5,8], [2,5,9], [2,6,8], [2,6,9]
• [3,5,7], [3,5,8], [3,5,9], [3,6,8], [3,6,9]

和最小的下降路徑是 [1,4,7]

提示：

1. 1 <= A.length == A[0].length <= 100
2. -100 <= A[i][j] <= 100

解題思路

class Solution:
    def minFallingPathSum(self, A):
        """
        :type A: List[List[int]]
        :rtype: int
        """
        if not A:
            return 0

        r, c = len(A), len(A[0])
        for row in
 
 range(r - 1, 0, -1):
            for col in range(c):
                if col == 0:
                    A[row - 1][col] += min(A[row][col], A[row][col + 1])
                elif col == c - 1:
                    A[row - 1][col] += min(A[row][col], A[row][col - 1])
                else:
                    A[
 
row - 1][col] += min(A[row][col], A[row][col + 1], A[row][col - 1])

        return min(A[0])

我們還有一種更簡潔的寫法。

class Solution:
    def minFallingPathSum(self, A):
        """
        :type A: List[List[int]]
        :rtype: int
        """
        while len(A) >= 2:
            row = A.pop()            
            for i in range(len(row)):
                A[-1][i] += min(row[max(0,i-1): min(len(row), i+2)])
        return min(A[0])

reference:

如有問題，希望大家指出！！！