Leetcode931.下降路径最小和

最新推荐文章于 2024-09-12 19:02:20 发布

歪比巴卜的棒棒糖

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量53

点赞数

文章标签：算法 leetcode 动态规划

本文链接：https://blog.csdn.net/gjvgjncg/article/details/134132899

版权

注意边界条件

class Solution {
public:
    int minFallingPathSum(vector<vector<int>>& matrix) {
        int n = matrix.size();
        vector<vector<int>> dp(n,vector<int>(n));
        for(int i=0;i<n;++i){
            dp[0][i] = matrix[0][i];
        }
        for(int i=1;i<n;++i){
            for(int j=0;j<n;++j){
                if(j==0) dp[i][j] = min(dp[i-1][j],dp[i-1][j+1]) + matrix[i][j];
                else if(j==n-1) dp[i][j] = min(dp[i-1][j],dp[i-1][j-1]) + matrix[i][j];
                else dp[i][j] = min(min(dp[i-1][j-1],dp[i-1][j+1]),dp[i-1][j]) + matrix[i][j];  
            }
            //for(int k =0;k<n;++k) cout<<dp[i][k]<<" ";
            cout<<endl;
        }
        return *min_element(dp[n-1].begin(),dp[n-1].end());
    }
};