力扣LeetCode: 931 下降路径最小和

题目:

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

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

示例 1:

复制代码
输入:matrix = [[2,1,3],[6,5,4],[7,8,9]]
输出:13
解释:如图所示,为和最小的两条下降路径

示例 2:

复制代码
输入:matrix = [[-19,57],[-40,-5]]
输出:-59
解释:如图所示,为和最小的下降路径

提示:

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

解法:动态规划

cpp 复制代码
class Solution {
public:
    // void print(vector<vector<int>> ans) {
    //     int n = ans.size();
    //     for (int i = 0; i < n; i++) {
    //         for (int j = 0; j < n; j++) {
    //             cout << ans[i][j] << " ";
    //         }
    //         cout << endl;
    //     }
    // }

    int minFallingPathSum(vector<vector<int>>& matrix) {
        int n = matrix.size();
        vector<vector<int>> ans(n, vector<int>(n, 0));
        // 初始第一行
        for (int i = 0; i < n; i++) {
            ans[0][i] = matrix[0][i];
        }

        for (int i = 1; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (j == 0) {
                    ans[i][j] = min(matrix[i][j] + ans[i - 1][j],
                                    matrix[i][j] + ans[i - 1][j + 1]);
                } else if (j == n - 1) {
                    ans[i][j] = min(matrix[i][j] + ans[i - 1][j],
                                    matrix[i][j] + ans[i - 1][j - 1]);
                } else {
                    ans[i][j] = min(matrix[i][j] + ans[i-1][j+1], min(matrix[i][j] + ans[i-1][j], matrix[i][j] + ans[i-1][j-1]));
                }
            }
        }
        return *min_element(ans[n-1].begin(), ans[n-1].end());
    }
};

代码解释

  1. 初始化

    • n 是矩阵的大小。

    • ans 是一个 n x n 的二维数组,用于存储从第一行到当前位置的最小路径和。

  2. 初始化第一行

    • 因为第一行的每个元素本身就是起点,所以 ans[0][i] 直接等于 matrix[0][i]
  3. 动态规划填充 ans 数组

    • 从第二行开始(i = 1),对于每一行的每个元素 matrix[i][j],计算从上一行的三个可能位置(左上、正上、右上)到当前位置的最小路径和。

    • 如果当前元素在第一列(j == 0),则只能从正上方或右上方移动过来。

    • 如果当前元素在最后一列(j == n - 1),则只能从正上方或左上方移动过来。

    • 否则,可以从左上方、正上方或右上方移动过来。

    • 通过 min 函数选择这三个可能路径中的最小值,并加上当前元素的值,更新 ans[i][j]

  4. 返回结果

    • 最终,ans 数组的最后一行存储了从第一行到最后一行的所有可能路径的最小和。

    • 使用 min_element 函数找到最后一行的最小值,并返回它。

复杂度分析

  • 时间复杂度O(n^2),因为需要遍历整个矩阵。

  • 空间复杂度O(n^2),因为需要存储 ans 数组。

相关推荐
岁忧3 分钟前
(LeetCode 面试经典 150 题) 138. 随机链表的复制 (哈希表)
java·c++·leetcode·链表·面试·go
重生之我是Java开发战士9 分钟前
【C语言】内存函数与数据在内存中的存储
c语言·开发语言·算法
君鼎12 分钟前
Effective C++ 条款17:以独立语句将newed对象置入智能指针
c++
极客BIM工作室1 小时前
深入理解C++中的Lazy Evaluation:延迟计算的艺术
开发语言·c++
roman_日积跬步-终至千里1 小时前
【机器学习】“回归“算法模型的三个评估指标:MAE(衡量预测准确性)、MSE(放大大误差)、R²(说明模型解释能力)
算法·机器学习·回归
kokunka1 小时前
【数据结构】栈的顺序存储(整型栈、字符栈)
数据结构
小指纹3 小时前
图论-最短路Dijkstra算法
数据结构·c++·算法·深度优先·图论
王德博客4 小时前
【从基础到实战】STL string 学习笔记(上)
c++·笔记·学习
赴3354 小时前
逻辑回归 银行贷款资格判断案列优化 交叉验证,调整阈值,下采样与过采样方法
算法·机器学习·逻辑回归·下采样·交叉验证·过采样·阈值
2501_924878734 小时前
无人机光伏巡检缺陷检出率↑32%:陌讯多模态融合算法实战解析
开发语言·人工智能·算法·视觉检测·无人机