leecode.931. 下降路径最小和

题目

下降路径最小和
给定一个方形整数数组 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。

思路

我们反着来,就是相当于我们求最小上升路径和.

代码

class Solution {
public:
    int minFallingPathSum(vector<vector<int>>& A) {
        int n = A.size();
        vector<vector<int>> dp(n + 1, vector<int>(n + 1, 0));
        for(int i = 1;i <= n;i++)
            dp[n][i] = A[n - 1][i - 1];
        for(int i = n - 1;i >= 1;i--){
            for(int j = 1;j <= n;j++){
                if(j == 1) dp[i][j] = min(dp[i + 1][j], dp[i + 1][j + 1]) + A[i - 1][j - 1];
                else if(j == n) dp[i][j] = min(dp[i + 1][j], dp[i + 1][j - 1]) + A[i - 1][j - 1];
                else dp[i][j] = min(dp[i + 1][j], min(dp[i + 1][j + 1], dp[i + 1][j - 1])) + A[i - 1][j - 1];
            }
        }
        int ans = 0x3f3f3f3f;
        for(int i = 1;i <= n;i++) ans = min(ans, dp[1][i]);
        return ans;
    }
};