给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。
思路:dp[i][j]表示到达点(i,j)的最小路径。点(i,j)不在两边时,dp[i][i]和上一层的两个结点有关;在两边时,只和上层的一个结点有关
int minimumTotal(vector<vector<int>>& triangle) {
int row=triangle.size();
vector<vector<int>>dp(row+1,vector<int>(row+1));
for(int i=1;i<=row;i++){
for(int j=1;j<=i;j++){
if(j==1){
dp[i][j]=dp[i-1][j]+triangle[i-1][j-1];
}
else if(j!=i){
dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+triangle[i-1][j-1];
}
else{
dp[i][j]=dp[i-1][j-1]+triangle[i-1][j-1];
}
}
}
for(int i=1;i<=row;i++){
dp[row][row]=min(dp[row][row],dp[row][i]);
}
return dp[row][row];
}
O(n)的空间复杂度
每一行的值都只和上一行有关,可以将空间复杂度优化为O(n),从前往后遍历会覆盖要用到的值,所以从后往前遍历
int minimumTotal(vector<vector<int>>& triangle) {
int row=triangle.size();
vector<int>dp(row+1);
for(int i=1;i<=row;i++){
for(int j=i;j>0;j--){
if(j==1){
dp[j]+=triangle[i-1][j-1];
}
else if(j!=i){
dp[j]=min(dp[j],dp[j-1])+triangle[i-1][j-1];
}
else{
dp[j]=dp[j-1]+triangle[i-1][j-1];
}
}
}
for(int i=1;i<=row;i++){
dp[row]=min(dp[row],dp[i]);
}
return dp[row];
}
巧妙的想法
倒着遍历,从三角形的最底部往上遍历,到dp[0]就是最小路径
int minimumTotal(vector<vector<int>>& triangle) {
int row=triangle.size();
vector<int>dp(triangle.back());
for(int i=row-2;i>=0;i--){
for(int j=0;j<=i;j++){
dp[j]=min(dp[j],dp[j+1])+triangle[i][j];
}
}
return dp[0];
}