给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
来源:力扣(LeetCode)
直接用dp:
class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
int n=triangle.size();
int [][]dp=new int[n][n];
dp[0][0]=triangle[0][0];
for(int i=1;i<n;i++){
dp[i][0]=triangle[i][0]+dp[i-1][0];
for(int j=1;j<i;j++){
dp[i][j]=triangle[i][j]+Math.min(dp[i-1][j-1],dp[i-1][j]);
}
dp[i][i]=triangle[i][i]+dp[i-1][i-1];
}
int res=dp[n-1][0];
for(int i=1;i<n;i++){
if(dp[n-1][i]<res){
res=dp[n-1][i];
}
}
return res;
}
}
上述算法使用了n平方的空间复杂度,还可以对空间复杂度进一步优化:
只用一个数组存储,从下往上遍历:
class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
int rows=triangle.size();
int []dp=new int[rows+1];//默认为0
for(int i=rows-1;i>=0;i--){
for(int j=0;j<=i;j++){
dp[j]=Math.min(dp[j],dp[j+1])+triangle.get(i).get(j);
}
}
return dp[0];
}
}