给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
思路一:首先使用递归的方式
画出递归树之后可以看到,假如上一层访问元素下标为i,那么当前层只能访问i,i+1。因此,通过传递当前和,层数,下标进行递归即可解决问题,最终当层数到达最底端,判断最小和即可。
时间复杂度:O(2n)
class Solution {
List<List<Integer>> triangle;
int res;
//sum是上一级传递的和,level是当前层次,i是上一层访问元素的下标
public void findNextLevel(int sum,int level,int i){
if(level==triangle.size()) {
res = Math.min(res,sum);
return;
}
findNextLevel(sum+triangle.get(level).get(i),level+1,i);
findNextLevel(sum+triangle.get(level).get(i+1),level+1,i+1);
}
public int minimumTotal(List<List<Integer>> triangle) {
this.triangle = triangle;
res = Integer.MAX_VALUE;
findNextLevel(triangle.get(0).get(0),1,0);
return res;
}
}
思路二:记忆化搜索
上面的方法虽然可以解决问题,但是一定会超时,通过观察递归树可以发现,存在着重复计算,所以要保存这些重复计算的量,来避免指数级时间复杂度。
但是要修改的时候,发现了我上面的方法,并不能修改为记忆化搜索,要做出改变,将递归函数的作为改为“返回在当前层该元素及以下的最小路径和”。
改变之后,再进行记忆化就简单了。
时间复杂度:O(n)
class Solution {
List<List<Integer>> triangle;
int[][] memo;
//sum是上一级传递的和,level是当前层次,i是上一层访问元素的下标
public int findNextLevel(int level,int i){
if(level==triangle.size()) return 0;
if(memo[level][i]==Integer.MAX_VALUE){
int left = findNextLevel(level+1,i);
int right = findNextLevel(level+1,i+1);
memo[level][i] = Math.min(left,right)+triangle.get(level).get(i);
}
return memo[level][i];
}
public int minimumTotal(List<List<Integer>> triangle) {
memo = new int[triangle.size()][triangle.size()];
for(int i=0;i<triangle.size();i++){
for(int j=0;j<triangle.size();j++){
memo[i][j] = Integer.MAX_VALUE;
}
}
this.triangle = triangle;
return findNextLevel(0,0);
}
}
思路三:动态规划
这就是三步走最后一步了,怎么将一个递归问题转化为自底向上的迭代。
class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
int length = triangle.size();
int[][] dp = new int[length+1][length+1];
for(int i=length-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];
}
}
思路四:一维数组动态规划
什么?这个题要用一维数组做?怎么做?
其实,每次我们在计算当前行的时候,只会用到下一行而已,再往下的就用不到了。所以每次用新的覆盖旧的就好了。
class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
int length = triangle.size();
int[] dp = new int[length+1];
for(int i=length-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];
}
}