题干
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
想法
动态规划
自底向上
有这样一个道理
就是如果从底向上 不管我怎么选一定是可以到达顶点的
也就是说只要自底向上的时候选的点是相邻即可
舍dp[i][j]是从底到i,j的最短路径
那么 dp[i][j] = Math.min(dp[i + 1][j], dp[i + 1][j + 1]) + triangle.get(i).get(j);
Java代码
public int minimumTotal(List<List<Integer>> triangle) {
if (triangle == null || triangle.size() == 0) {
return 0;
}
int[][] dp = new int[triangle.size()][triangle.size()];
for (int j = triangle.size() - 1; j >= 0; j--) {
dp[triangle.size() - 1][j] = triangle.get(triangle.size() - 1).get(j);
}
for (int i = triangle.size() - 2; i >= 0; i--) {
for (int j = 0; j < triangle.get(i).size(); j++) {
dp[i][j] = Math.min(dp[i + 1][j], dp[i + 1][j + 1]) + triangle.get(i).get(j);
}
}
return dp[0][0];
}
另一种更巧妙的办法
使用一维数组int[]来存储每一行
原因是当我用i行时,其实i+1行的数据已经不再重要
public int minimumTotal(List<List<Integer>> triangle) {
int row = triangle.size();
int[] minlen = new int[row+1];
for (int level = row-1;level>=0;level--){
for (int i = 0;i<=level;i++){ //第i行有i+1个数字
minlen[i] = Math.min(minlen[i], minlen[i+1]) + triangle.get(level).get(i);
}
}
return minlen[0];
}