题目
给定一个三角形 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
思路
方法是从三角形的底部向上进行计算。它避免了在顶部处理只有一行的情况,因为循环是从 triangle.size() - 2
开始的,这意味着它不会执行循环体。
-
vector<int> dp(triangle.back());
创建一个dp
数组,其初始值是三角形的最后一行。这是因为最后一行的元素就是它们自己的路径和。 -
for(int i = triangle.size() - 2; i >= 0; i --)
是一个从倒数第二行开始,向上遍历三角形的循环。 -
for(int j = 0; j <= i; j ++)
是一个遍历当前行的循环。注意这里j
的上限是i
,这是因为三角形的每一行的长度是递减的。 -
dp[j] = min(dp[j], dp[j + 1]) + triangle[i][j];
这一行是动态规划的核心。它计算到达当前位置的最小路径和,考虑从当前位置的下方或右下方到达的最小路径和,并加上当前位置的值。 -
return dp[0];
返回dp
数组的第一个元素,这是到达三角形顶部(即(0,0)
位置)的最小路径和
代码
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int rows = triangle.size();
int cols = triangle[rows - 1].size();
// vector<int> dp(rows, 0);
vector<int> dp(triangle.back());
//triangle.back() 是一个成员函数,它用于访问 vector<vector<int>>
// 类型的最后一个元素。
// triangle 是一个二维向量,其中包含多个向量,每个向量代表一行。
int j = 0;
for (int i = rows-2; i>=0; i--)
{
for (int j = 0; j <= i; j++)
{
dp[j] = min(dp[j], dp[j + 1]) + triangle[i][j];
}
}
return dp[0];
}
};