leetcode 931.下降路径最小和

931.下降路径最小和

文章目录

  • 931.下降路径最小和
  • 一、题目
    • 1.题目描述
    • 2.基础框架
    • 3.解题思路

一、题目

原题链接:931.下降路径最小和

1.题目描述

给你一个 n x n方形 整数数组 matrix ,请你找出并返回通过 matrix下降路径最小和

下降路径 可以从第一行中的任何元素开始,并从每一行中选择一个元素。在下一行选择的元素和当前行所选元素最多相隔一列(即位于正下方或者沿对角线向左或者向右的第一个元素)。具体来说,位置 (row, col) 的下一个元素应当是 (row + 1, col - 1)(row + 1, col) 或者 (row + 1, col + 1)

示例 1:

leetcode 931.下降路径最小和_第1张图片
输入:matrix = [[2,1,3],[6,5,4],[7,8,9]]
输出:13
解释:如图所示,为和最小的两条下降路径

示例 2:

leetcode 931.下降路径最小和_第2张图片
输入:matrix = [[-19,57],[-40,-5]]
输出:-59
解释:如图所示,为和最小的下降路径

提示:

  • n == matrix.length == matrix[i].length
  • 1 <= n <= 100
  • -100 <= matrix[i][j] <= 100

2.基础框架

C++基础框架代码如下:

int minFallingPathSum(vector<vector<int>>& matrix) {
}

3.解题思路

  • 题目分析
  1. 题目目标是求下降路径的最小和,属于最小值问题。

  2. 从第i行中的任意元素只能向下的方向求和。

  3. (i,j)方块可以下降到(i,j - 1)(i, j)(i,j + 1)的方块。

  4. 可以得出递推公式为:

    martrix[i][j] += min{martrix[i - 1][j - 1], martrix[i - 1][j], martrix[i - 1][j + 1]}

  5. j = 0时或者j = matrix.size() - 1时,需要另外处理:

    • j = 0,则只能从上一层的jj + 1过来。
    • j = matrix.size() - 1,则只能从上一层的j - 1j过来。
    • 假如只存在j = 1的matrix数组,即j = matrix.size() - 1 = 0,那么只能从上一层的j走下来。
  6. 最后一行进行一次排序,选出最小的路径和。

  • 实现代码:

    int minFallingPathSum(vector<vector<int>>& matrix) {
        int i, j;
        for (i = 1; i < matrix.size(); i++)
        {
            for (j = 0; j < matrix[i].size(); j++)
            {
                if (j == 0 && j == matrix[i].size() - 1) matrix[i][j] += matrix[i - 1][j];
                else if (j == 0) matrix[i][j] += min(matrix[i - 1][j], matrix[i - 1][j + 1]);
                else if (j == matrix[i].size() - 1) matrix[i][j] += min(matrix[i - 1][j - 1], matrix[i - 1][j]);
                else
                    matrix[i][j] += min(matrix[i - 1][j - 1], min(matrix[i - 1][j], matrix[i - 1][j + 1]));
            }
        }
        sort(matrix[i - 1].begin(), matrix[i - 1].end());
        return matrix[i - 1][0];
    }
    

你可能感兴趣的:(算法,算法)