题目
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
解答
要求最小路径就要先求出第n行以每个元素结尾的最小路径,以第n行的第m个元素结尾的最小路径为f(m,n) = min{f(n-1,m-1), f(n-1,m)},因此我们可以用一个长度为行数的数组保存到达每个元素的最小路径。(注意其中tmp变量作为临时存储当前行更新的最短路径)
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
// f(n, m) = min{f(n-1,m-1), f(n-1,m)}
int n = triangle.size();
vector<int> dp(n);
dp[0] = triangle[0][0];
for (int i = 1; i < n; i++) {
vector<int> tmp(i+1);
for (int j = 0; j <= i; j++) {
if (j == 0) {
tmp[j] = dp[j] + triangle[i][j];
}
else if (j == i) {
tmp[j] = dp[j-1] + triangle[i][j];
}
else {
tmp[j] = min(dp[j-1], dp[j]) + triangle[i][j];
}
}
for (int i = 0; i < tmp.size(); i++)
dp[i] = tmp[i];
}
int res = dp[0];
for (int i = 1; i < dp.size(); i++) {
if (res > dp[i])
res = dp[i];
}
return res;
}
};