题目:
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
题解思路:
方法一:二维动态规划,自顶向下 含优化版本
1.初始化第一行dp[0][0]=triangle[0][0];
2.状态转移方程:当j=0时,dp[i][0]=triangle[i][j]+dp[i-1][j];
当j=triangle[i].size()-1时,也就是第i行的最后一个元素时,
dp[i][j]=triangle[i][j]+dp[i-1][j-1];
当j介于两者之间,dp[i][j]=min(d[[i-1][j-1],dp[i-1][j])+triangle[i][j];
3.求最小值,记得在最后一行去求,i从0到n-1,dp[m-1][i] 最小值就是最短路径之和。因为最后不是dp[m-1][n-1],所以需要最后遍历一下。
函数代码:
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int m=triangle.size();
int n=triangle[m-1].size();
vector<vector<int>>dp(m,vector<int>(m,0));
dp[0][0]=triangle[0][0];
for(int i=1;i<m;i++)
{
for(int j=0;j<triangle[i].size();j++)
{
if(j==0)
{
dp[i][j]=triangle[i][j]+dp[i-1][j];
}
else if(j==triangle[i].size()-1)
{
dp[i][j]=triangle[i][j]+dp[i-1][j-1];
}
else
{
dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j];
}
}
}
int res=INT_MAX;
for(int i=0;i<n;i++)
{
res=min(res,dp[m-1][i]);
}
return res;
}
};
代码优化后:
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int m=triangle.size();
int n=triangle[m-1].size();
vector<vector<int>>dp(m,vector<int>(m,0));
dp[0][0]=triangle[0][0];
for(int i=1;i<m;i++)
{
dp[i][0]=dp[i-1][0]+triangle[i][0];
for(int j=1;j<i;j++)
{
dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j];
}
dp[i][i]=triangle[i][i]+dp[i-1][i-1];
}
int res=INT_MAX;
for(int i=0;i<n;i++)
{
res=min(res,dp[m-1][i]);
}
return res;
}
};