题目:
Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.
思路:
直观思路就是递归遍历所有路径,但是submit显示超时。我测试了几个用例,结果都是对的。所以思路是对的,但效率太低太低。
然后想了一下,里面好像存在大量的重复运算。所以,利用动态规划的方法来消除重复的运算。
代码实现:
超时的代码:
class Solution {
public:
void f(vector<vector<int>>& triangle, int i, int last_ii, int sum, int &min){
if (i >= triangle.size()){
if (sum < min){
min = sum;
}
return ;
}
f(triangle, i+1, last_ii, sum + triangle[i][last_ii], min);
f(triangle, i+1, last_ii+1, sum + triangle[i][last_ii+1], min);
}
int minimumTotal(vector<vector<int>>& triangle) {
int min = INT_MAX;
f(triangle, 1, 0, triangle[0][0], min);
return min;
}
};
动态规划:
从triangle的倒数第二行开始算,一直算到第一行。直接在原数组上修改即可。
动态规划的公式:triangle[i][j] += min(triangle[i+1][j], triangle[i+1][j+1])
,triangle[i][j]表示第i行的第j个元素。
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
for (int i = triangle.size()-2; i >= 0; --i){
for (int j = 0; j < triangle[i].size(); ++j){
triangle[i][j] += min(triangle[i+1][j], triangle[i+1][j+1]);
}
}
return triangle[0][0];
}
};
discuss:
思路:从倒数第二层开始,自下而上。
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n = triangle.size();
vector<int> minlen(triangle.back()); // 最后一层
for (int layer = n - 2; layer >= 0; --layer){ // 刨除最后一层的其它层(bottom to top)
for (int i = 0; i <= layer; ++i){ // 遍历一层中的每个node
minlen[i] = min(minlen[i], minlen[i+1]) + triangle[layer][i]; // 用layer+1层求layer层(比如最后两层:6对应4和1,5对应1和8,7对应8和3)
}
}
return minlen[0];
}
};