三角形最小路径和
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
思路和代码
动态规划,首先定义一个dp数组,数组的大小和三角形最后一行大小一致,dp[0] = triangle[0][0],然后遍历每一行,因为只使用一个数组,所以从后往前遍历,对于最后一个元素和第一个元素要单独处理。
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int rows = triangle.size();
int cols = triangle[rows-1].size();
vector<int> dp(cols);
dp[0] = triangle[0][0];
for(int i=1; i<rows; i++){
int col = triangle[i].size()-1;
dp[col] = dp[col-1] + triangle[i][col];
col--;
for(; col>0; col--){
dp[col] = min(dp[col], dp[col-1]) + triangle[i][col];
}
dp[col] = dp[col] + triangle[i][col];
}
int res = dp[0];
for(int i=1; i<cols; i++){
res = min(res, dp[i]);
}
return res;
}
};