leetcode 931. 下降路径最小和(Minimum Falling Path Sum)

目录

• 题目描述：
• 示例：
• 解法：

题目描述：

给定一个方形整数数组 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]，所以答案是 12。

提示：

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

---

解法：

class Solution {
public:
    int minFallingPathSum(vector<vector<int>>& A) {
        int m = A.size();
        int n = A[0].size();
        vector<int> cur = A[0];
        vector<int> pre = A[0];
        for(int i = 1; i < m; i++){
            if(n > 1){
                cur[0] = min(pre[0], pre[1]) + A[i][0];
                cur[n-1] = min(pre[n-2], pre[n-1]) + A[i][n-1];
            }else{
                cur[0] += A[i][0];
            }
            for(int j = 1; j < n-1; j++){
                int min_val = min(pre[j], pre[j-1]);
                min_val = min(pre[j+1], min_val);
                cur[j] = min_val + A[i][j];
            }
            pre = cur;
            // for(int val : cur){
            //     cout<<val<<", ";
            // }
            // cout<<endl;
        }
        int res= INT_MAX;
        for(int val : cur){
            res = min(res, val);
        }
        return res;
    }
};