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;
}
};