leetcode 1289. 下降路径最小和 II（C++）

给你一个整数方阵 arr ，定义「非零偏移下降路径」为：从 arr 数组中的每一行选择一个数字，且按顺序选出来的数字中，相邻数字不在原数组的同一列。

请你返回非零偏移下降路径数字和的最小值。

 

示例 1：

输入：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

C++

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