题目大意
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分
解题思路
方法一:
很明显这道题可以通过递归完成,而递归可以改写成动态规划,因此我们直接写动态规划即可。
对于任意位置[i][j]
,想要到达该位置,出发位置只能是[i-1][j-1]
和[i-1][j]
。因此到[i][j]
位置的最小路径仅取决于上述两个位置,找到二者的小值即可。
另外,每行的第一个元素仅能通过上一行第一个元素到达,每行最后一个元素仅能通过上一行最后一个元素到达。
因此可以创建一个2D 的dp表,每行的有效范围是0~row-1.
该方法使用了2D dp表,浪费空间,就不写具体代码了。
方法二:
每一行的取值仅取决于上一行的取值,因此可以采用1D的dp数组完成上述操作。另外需要注意的是,在求解第i层的最短路径时,需要从后向前遍历,且该dp数组的有效区间为0~i.
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
if(triangle.size() == 0){
return 0;
}
int length = triangle.size();
vector<int> dp(legth, INT_MAX);
dp[0] = triangle[0][0];
// 外层循环是依次求解每一行的最短路径
for (int i = 1; i < length; ++i){
// 当求解第i层最短路径时,dp的有效范围是0~i
for (int j = i; j >= 0; --j){
// 第i层的最后一个元素仅取决于上一行的最后一个元素
if (j == i){
dp[j] = dp[j - 1] + triangle[i][j];
}
// 第i层的首元素仅取决于上一行的首元素
else if (j == 0){
dp[j] = dp[j] + triangle[i][j];
}
else{
dp[j] = min(dp[j], dp[j - 1]) + triangle[i][j];
}
}
}
// 遍历一遍找到最大值
int res = INT_MAX;
for (int i = 0; i < length; ++i){
if(dp[i] < res)
res = dp[i];
}
return res;
}
};