120. 三角形最小路径和
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
示例 1:
输入:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
输出: 11
思路分析
一道很基础的dp题,假设dp[i][j]代表在第i行,第j列时取得的最小值
那么状态转移方程可以写为
dp[i][j] =min(dp[i-1][j-1],dp[i-1][j]) + triangle[i][j];
同时注意边界问题,从上至下不断更新dp,最后从dp的最后一行选取最小值即为正确结果。
通过代码
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int m = triangle.size();
vector<vector<int>> dp(m,vector<int>(m));
dp[0][0] = triangle[0][0];
for (int i=1;i<m;i++){
for(int j=0;j<=i;j++){
if (j==0){
dp[i][j] = dp[i-1][j] + triangle[i][j];
}
else if (i==j){
dp[i][j] = dp[i-1][j-1] + triangle[i][j];
}
else
dp[i][j] =min(dp[i-1][j-1],dp[i-1][j]) + triangle[i][j];
}
}
int min = INT_MAX;
for (int i=0;i<m;i++){
if (dp[m-1][i]<min){
min = dp[m-1][i];
}
}
return min;
}
};
通过滚动数组方式可以减少空间复杂度