题目
给定一个三角形 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)
分析
我们自上而下处理的时候,很明显每一行的第一个元素的dp公式可以写成dp[i][j] = dp[i-1][j] + triangle[i][j],而最后一行可以写成dp[i][j] = dp[i-1][j-1] + triangle[i][j]。剩下的中间元素就是不确定元素了,因为剩下的这些元素要么可以从上一行的前一列得来,要么从上一行的当前列得来,所以这里就需要注意怎么比较不同来源的元素大小(这道题目花了博主3个晚上的时间,真的是不容易)
import java.util.List;
import java.util.ArrayList;
public class Triangle {
public static void main(String[] args) {
List<List<Integer>> lis = new ArrayList();
List<Integer> aa = new ArrayList();
aa.add(2);
lis.add(aa);
aa = new ArrayList();
aa.add(3);
aa.add(4);
lis.add(aa);
aa = new ArrayList();
aa.add(6);
aa.add(5);
aa.add(7);
lis.add(aa);
aa = new ArrayList();
aa.add(4);
aa.add(1);
aa.add(8);
aa.add(3);
lis.add(aa);
System.out.println(minimumTotal(lis));
}
public static int minimumTotal(List<List<Integer>> triangle) {
int row = triangle.size();
int col = triangle.get(row-1).size();
int[][] dp = new int[row][col];
dp[0][0] = triangle.get(0).get(0);
for(int i = 1;i<row;i++) {
for(int j = 0;j<triangle.get(i).size();j++) {
if(j == triangle.get(i).size() - 1) {
dp[i][j] = dp[i-1][j-1] + triangle.get(i).get(j);
} else if(j == 0) {
dp[i][j] = dp[i-1][j] + triangle.get(i).get(j);
} else {
dp[i][j] = Math.min(dp[i][j],dp[i-1][j] + triangle.get(i).get(j));
}
if(j < triangle.get(i).size() - 1) {
dp[i][j+1] = dp[i-1][j] + triangle.get(i).get(j+1);
}
}
}
int mx = Integer.MAX_VALUE;
for(int j = 0;j<col;j++) {
mx = Math.min(mx,dp[row-1][j]);
}
return mx;
}
}