也是字节跳动遇到的面试题
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
例如,给定三角形:
[ [2], [3,4], [6,5,7], [4,1,8,3] ]
自顶向下的最小路径和为 11
(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
int[][] res = new int[triangle.size()][triangle.size()];//存List数据
//动态规划表
int[][] dp = new int[triangle.size()][triangle.get(triangle.size() - 1).size()];
int i = 0, j = 0;
for (List<Integer> a : triangle) {
j = 0;
for (Integer b : a) {
res[i][j++] = b;
}
i++;
}
//和以前做过的不同,这次动态规划表是从后往前写,遍历也是从后往前
//初始化最后一行
for (int a = 0; a < res[res.length - 1].length; a++) {
dp[res.length - 1][a] = dp[res.length - 1][a] + res[res.length - 1][a];
}
//逐层向上遍历
for (int a = res.length - 2; a >= 0; a--) {
for (int b = 0; b < res[a].length - 1; b++) {
dp[a][b] = Math.min(dp[a + 1][b], dp[a + 1][b + 1]) + res[a][b];
}
}
for (int a = 0; a < triangle.size(); a++) {
for (int b = 0; b < triangle.get(a).size(); b++) {
System.out.print(dp[a][b] + " ");
}
System.out.println();
}
return dp[0][0];
}
}