给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是下标与上一层结点下标相同或者等于上一层结点下标 + 1 的两个结点。
也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 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)。
思路
最值问题一般用动态规划解决。而且题目中的状态转移方程已经非常明确了。
dp[i][j]=min(dp[i-1][j],dp[i-1][j-1])+nums[i][j]
- 代码
public int minimumTotal(List<List<Integer>> triangle) {
int n = triangle.size();
//表示到达第i行第j列的最小路径
int[][] dp = new int[n][n+1];
for (int i = 0; i < n; i++) {
Arrays.fill(dp[i],Integer.MAX_VALUE);
}
//初始化,第一行第一列的值是固定的
for (int i = 1; i <= triangle.get(0).size(); i++) {
dp[0][i] = triangle.get(0).get(i - 1);
}
for (int i = 1; i < n; i++) {
for (int j = 1; j <= triangle.get(i).size(); j++) {
//dp[i][j]只能由上一行的dp[i-1][j]或dp[i-1][j-1]得到
dp[i][j] = Math.min(dp[i-1][j],dp[i-1][j-1])+triangle.get(i).get(j-1);
}
}
//返回最后一行的最小值
return Arrays.stream(dp[n-1]).min().getAsInt();
}
- 在评论区中看到一个不用初始化的代码,而且比较易懂。上面的代码是自顶向下遍历,下面的是自底向上的。
public int minimumTotal1(List<List<Integer>> triangle) {
if (triangle.size()==0 || triangle==null)
return 0;
int n = triangle.size();
int[][] dp = new int[n + 1][n + 1];
//自底向上遍历
for (int i = n-1;i>=0;i--){
List<Integer> list = triangle.get(i);
for (int j = 0; j < list.size(); j++) {
dp[i][j] = Math.min(dp[i+1][j],dp[i+1][j+1])+list.get(j);
}
}
return dp[0][0];
}