动态规划-下降路径最小和

题目描述

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

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

解题思路

这个问题是一个典型的动态规划(Dynamic Programming, DP)问题。动态规划通常用于解决那些可以通过将原问题分解为较小子问题,然后解决这些子问题来解决整个问题的情况。在这个问题中,我们可以从最后一行开始向上回溯,计算到达每一行每一列的最小路径和。

步骤
  1. 初始化
    • 创建一个同样大小的DP数组(dp),用于存储到达每个位置的最小路径和。
    • 将DP数组的第一行初始化为矩阵的第一行,因为从第一行开始没有选择,只能取矩阵本身的值。
  2. 填充DP数组
    • 从第二行开始,遍历矩阵的每一行和每一列。
    • 对于每个位置 (i, j)i 表示行,j 表示列),计算从下一行 (i+1, j-1)(i+1, j)(i+1, j+1) 中选择的最小路径和,并将这个最小路径和加上当前位置的值 matrix[i][j],存入 dp[i][j]
    • 注意边界条件:当 j-1 < 0j+1 >= n 时,需要忽略不存在的位置。
  3. 找出最小值
    • 遍历DP数组的最后一行,找出其中的最小值,即为所求的最小路径和。
cpp 复制代码
class Solution {
public:
    int minFallingPathSum(vector<vector<int>>& matrix) {
        int n = matrix.size();
        vector<vector<int>> dp(n, vector<int>(n, 0));
        // 初始化dp数组的第一行为matrix的第一行
        for (int j = 0; j < n; ++j) {
            dp[0][j] = matrix[0][j];
        }

        for (int i = 1; i < n; i++) {
            for (int j = 0; j < n; ++j) {
                // 计算从下一行到当前位置的最小路径和
                int minNext = INT_MAX;
                // 考虑从正下方和左右对角线的下一个位置(如果存在)
                for (int k = max(0, j - 1); k <= min(n - 1, j + 1); ++k) {
                    minNext = min(minNext, dp[i - 1][k]);
                }
                // 更新dp[i][j]为当前位置的值加上从下一行到当前位置的最小路径和
                dp[i][j] = matrix[i][j] + minNext;
            }
        }
        int ret = dp[n - 1][0];
        for (int i = 1; i < n; i++) {
            if (ret >= dp[n - 1][i])
                ret = dp[n - 1][i];
        }
        return ret;
    }
};

复杂度分析

  • 时间复杂度:O(n^2),其中 n 是矩阵的维度。因为我们需要遍历矩阵的每个元素一次来填充DP数组,并再次遍历DP数组的最后一行来找出最小值。
  • 空间复杂度:O(n^2),用于存储DP数组。
相关推荐
雾月5515 分钟前
LeetCode 1292 元素和小于等于阈值的正方形的最大边长
java·数据结构·算法·leetcode·职场和发展
知来者逆2 小时前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊2 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北2 小时前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈3 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing3 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD3 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘5 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV5 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8246 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode