给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。
示例 1:
输入:triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出:11
解释:如下面简图所示:
2
3 4
6 5 7
4 1 8 3
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/triangle
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解一、迭代+存储求解,迭代过程记录求过的结果,避免遇到相同索引再去迭代。
class Solution{
Integer[][] memo;
public int minimumTotal(List<List<Integer>> triangle){
memo = new Integer[triangle.size()][triangle.size()];
return dfs(triangle, 0, 0);
}
public int dfs(List<List<Integer>> triangle, int i, int j){
if(i == triangle.size()){
return 0;
}
if(memo[i][j] != null){
return memo[i][j];
}
return memo[i][j] = Math.min(dfs(triangle, i + 1, j), dfs(triangle, i + 1, j + 1)) + triangle.get(i).get(j);
}
}
题解二、dp,建立triangle.size() + 1维数组,存储求和数据,由下至顶进行求和找最小。
class Solution{
public int minimumTotal(List<List<Integer>> triangle){
int n = triangle.size();
int[][] dp = new int[n + 1][n + 1];
for(int i = n - 1; i >= 0; i--){
for(int j = 0; j <= i; j++){
dp[i][j] = Math.min(dp[i + 1][j], dp[i + 1][j + 1]) + triangle.get(i).get(j);
}
}
return dp[0][0];
}
}
题解三、空间优化,dp定义为一维数组,随行进行更新dp[j]和dp[j + 1]即可完成求和过程中最小的查询选择。(之前二维记录的是各行的最小求和)
class Solution{
public int minimumTotal(List<List<Integer>> triangle){
int n = triangle.size();
int[] dp = new int[n + 1];
for(int i = n - 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];
}
}