三角形最小路径和(动态规划)

6 篇文章 0 订阅
5 篇文章 0 订阅

120.三角形最小路径和
给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。

示例 1:
输入:triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出:11
解释:如下面简图所示:

   2
  3 4
 6 5 7
4 1 8 3

自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。

示例 2:
输入:triangle = [[-10]]
输出:-10

提示:
1 <= triangle.length <= 200
triangle[0].length == 1
triangle[i].length == triangle[i - 1].length + 1
-104 <= triangle[i][j] <= 104

思路:动态规划寻找每一步的最小值
将这三角形想象成:(第i行就有i个数)

2
3 4
6 5 7
4 1 8 3

动态规划步骤:
1.找初始状态

dp[0][0]=triangle[0][0];

2.确定循环顺序:都是从小到大(外层是行数,内层是遍历该行的每个数)
3 状态转移方程式:
分三种情况:
(1)每一行的第一个数只能是该数的值加上上一行的第一个dp[i-1][0]

dp[i][0]=dp[i-1][0]+triangle[i][0];

(2)每一行的最后一个数只能是该数的值加上上一行的最后一个dp[i-1][i-1]

dp[i][i]=dp[i-1][i-1]+triangle[i][i];

(3)每一行的中间的每个数都在内层循环遍历

dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j];

时间复杂度O(n2)
空间复杂度O(n2)(二维数组)

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        int n=triangle.size();
        vector<vector<int>> dp(n, vector<int>(n));
        dp[0][0]=triangle[0][0];
        for(int i=1;i<n;i++)
        {
            dp[i][0]=dp[i-1][0]+triangle[i][0];//遍历每层数前先要求出第一条路径
            for(int j=1;j<i;j++)
            {
                dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j];
            }
            dp[i][i]=dp[i-1][i-1]+triangle[i][i];
        }
        int mid=dp[n-1][0];
        for(int i=1;i<n;i++)
        {
            mid=min(mid,dp[n-1][i]);
        }
        return mid;
    }
};

空间复杂度有点高,需要优化一下:
观察状态转移方程式

dp[i][0]=dp[i-1][0]+triangle[i][0];
dp[i][i]=dp[i-1][i-1]+triangle[i][i];
dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j];

这里的dp都只与上一层的i-1有关系,所以我们不需要保存更上一层,所以可以仿照背包问题的优化,改成一维的滚动数组:
1.初始状态:dp[0]=triangle[0][0];
2.确定循环顺序:外层从小到大遍历,内层从大到小遍历该行所有数
内层为什么要从大到小遍历?
观察内层循环的状态转移方程:dp[j]=min(dp[j],dp[j-1])+triangle[i][j];
因为每次遍历都要选出该位置dp[j]和上一位置dp[j-1]中较小的位置,如果从左到右遍历,那么会覆盖下一次循环要用到的dp[j-1],下一次循环要用的dp[j-1]就是本层循环的dp[j](被覆盖,不是想要的dp[j-1]),所以不能从小到大循环,从大到小循环想要用的dp[j-1]还未被赋值(覆盖),所以可以实现滚动数组。

3.确定状态转移方程式:

dp[0]+=triangle[i][0];
dp[i]=dp[i-1]+triangle[i][i];
dp[j]=min(dp[j],dp[j-1])+triangle[i][j];

时间复杂度O(n2)
空间复杂度O(n1)

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        int n=triangle.size();
        vector<int>dp(n);
        dp[0]=triangle[0][0];
        for(int i=1;i<n;i++)
        {
            dp[i]=dp[i-1]+triangle[i][i];//由于循环交换了顺序遍历每层前先求出最后一条路径
            for(int j=i-1;j>0;j--)
            {
                dp[j]=min(dp[j],dp[j-1])+triangle[i][j];
            }
            dp[0]+=triangle[i][0];
        }
        int mid=dp[0];
        for(int i=1;i<n;i++)
        {
            mid=min(mid,dp[i]);
        }
        return mid;
    }
};
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值