Leetcode 1289.下降路径最小和 II
1 题目描述(Leetcode题目链接)
给你一个整数方阵 arr ,定义「非零偏移下降路径」为:从 arr 数组中的每一行选择一个数字,且按顺序选出来的数字中,相邻数字不在原数组的同一列。
请你返回非零偏移下降路径数字和的最小值。
输入:arr = [[1,2,3],[4,5,6],[7,8,9]]
输出:13
解释:
所有非零偏移下降路径包括:
[1,5,9], [1,5,7], [1,6,7], [1,6,8],
[2,4,8], [2,4,9], [2,6,7], [2,6,8],
[3,4,8], [3,4,9], [3,5,7], [3,5,9]
下降路径中数字和最小的是 [1,5,7] ,所以答案是 13 。
提示:
- 1 <= arr.length == arr[i].length <= 200
- -99 <= arr[i][j] <= 99
2 题解
动态规划,直接在原数组上操作,一个元素应加上上一行的最小值,如果最小值刚好在它头上
,那就加上上一行的次小值。
class Solution:
def minFallingPathSum(self, arr: List[List[int]]) -> int:
n = len(arr)
for i in range(1, n):
a, b = sorted(arr[i-1])[:2] # 上一层的两个最小的值
for j in range(n):
if arr[i-1][j] == a:
arr[i][j] += b
else:
arr[i][j] += a
return min(arr[n-1])