题目链接:https://leetcode-cn.com/problems/minimum-falling-path-sum/
题意:
给你一个 n x n 的 方形 整数数组 matrix ,请你找出并返回通过 matrix 的下降路径 的 最小和 。
下降路径 可以从第一行中的任何元素开始,并从每一行中选择一个元素。在下一行选择的元素和当前行所选元素最多相隔一列(即位于正下方或者沿对角线向左或者向右的第一个元素)。具体来说,位置 (row, col) 的下一个元素应当是 (row + 1, col - 1)、(row + 1, col) 或者 (row + 1, col + 1) 。
方法:动态规划,在原来的数组基础上,构造动态规划数组,通过最后一行找到最短路径
class Solution {
public:
int minFallingPathSum(vector<vector<int>>& matrix) {
int size = matrix.size();//记录矩阵长度
for(int i=1;i<size;i++)//枚举每一行
{
for(int j=0;j<size;j++)//枚举每一列
{
if(j==0)//如果是第一列
{
matrix[i][j] = min(matrix[i-1][j],matrix[i-1][j+1])+matrix[i][j];//第一列,需要考虑上面和右上角哪个值更小,更新当前节点的dp值
}else if(j==size-1)
{
matrix[i][j] = min(matrix[i-1][j],matrix[i-1][j-1])+matrix[i][j];//最后一列,需要考虑上面和左上角哪个值更小
}else
{
int tmp = min(matrix[i-1][j],matrix[i-1][j-1]);
matrix[i][j] = min(tmp,matrix[i-1][j+1])+matrix[i][j];//否则,需要考虑上面,左上角和右上角哪个值更小
}
}
}
int ans = matrix[size-1][0];//记录最小下降和
for(int i=1;i<size;i++) ans = min(ans,matrix[size-1][i]);//不断更新最小下降和
return ans;
}
};