1.题目
2.题解
2.1动态规划
2.2Dp+优化1
2.3Dp+优化3
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
示例:
给定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
说明:
如果你可以只使用 O(n) 的额外空间(n 为三角形的总行数)来解决这个问题,那么你的算法会很加分。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/triangle
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
用dp[i][j]表示走到位置(i,j)的最小路径和,即三角形的第i行第j列的位置(均从0开始),且第i行有i+1个元素。
根据题设约束,状态转移方程如下:
public class Solution120 {
@Test
public void test120() {
List> triangle = new ArrayList<>();
List list1 = Collections.singletonList(2);
List list2 = Arrays.asList(3, 4);
List list3 = Arrays.asList(6, 5, 7);
List list4 = Arrays.asList(4, 1, 8, 3);
triangle.add(list1);
triangle.add(list2);
triangle.add(list3);
triangle.add(list4);
System.out.println(minimumTotal(triangle));
}
public int minimumTotal(List> triangle) {
int n = triangle.size();
int[][] dp = new int[n][n];
dp[0][0] = triangle.get(0).get(0);
for (int i = 1; i < n; i++) {
dp[i][0] = dp[i - 1][0] + triangle.get(i).get(0);
for (int j = 1; j < i; j++) {
dp[i][j] = Math.min(dp[i - 1][j - 1], dp[i - 1][j]) + triangle.get(i).get(j);
}
dp[i][i] = dp[i - 1][i - 1] + triangle.get(i).get(i);
}
int minSum = dp[n - 1][0];
for (int t : dp[n - 1]) {
minSum = Math.min(minSum, t);
}
return minSum;
}
}
从2.1状态转移方程可知,每轮运算只与上一轮的结果有关,故不用存储上一轮之前的结果。只用开辟两个长度为n的数组,交替使用即可。
public class Solution120 {
@Test
public void test120() {
List> triangle = new ArrayList<>();
List list1 = Collections.singletonList(2);
List list2 = Arrays.asList(3, 4);
List list3 = Arrays.asList(6, 5, 7);
List list4 = Arrays.asList(4, 1, 8, 3);
triangle.add(list1);
triangle.add(list2);
triangle.add(list3);
triangle.add(list4);
System.out.println(minimumTotal(triangle));
}
public int minimumTotal(List> triangle) {
int n = triangle.size();
int[][] dp = new int[2][n];
dp[0][0] = triangle.get(0).get(0);
for (int i = 1; i < n; ++i) {
int curr = i % 2;
int prev = 1 - curr;
dp[curr][0] = dp[prev][0] + triangle.get(i).get(0);
for (int j = 1; j < i; ++j) {
dp[curr][j] = Math.min(dp[prev][j - 1], dp[prev][j]) + triangle.get(i).get(j);
}
dp[curr][i] = dp[prev][i - 1] + triangle.get(i).get(i);
}
int minSum = dp[(n - 1) % 2][0];
for (int t : dp[(n - 1) % 2]) {
minSum = Math.min(minSum, t);
}
return minSum;
}
}
由于后面的行都比前面行元素多1;且当计算第i行第j列时,只需要第i-1行第j-列的结果和第i-1行第j列的结果,如果让j从i递减到0,即结果数组逆序填充,则前面的值在使用后才被填充,不会影响结果,故可进一步优化空间为一个n长度数组。
public class Solution120 {
@Test
public void test120() {
List> triangle = new ArrayList<>();
List list1 = Collections.singletonList(2);
List list2 = Arrays.asList(3, 4);
List list3 = Arrays.asList(6, 5, 7);
List list4 = Arrays.asList(4, 1, 8, 3);
triangle.add(list1);
triangle.add(list2);
triangle.add(list3);
triangle.add(list4);
System.out.println(minimumTotal(triangle));
}
public int minimumTotal(List> triangle) {
int n = triangle.size();
int[] dp = new int[n];
dp[0] = triangle.get(0).get(0);
for (int i = 1; i < n; ++i) {
dp[i] = dp[i - 1] + triangle.get(i).get(i);
for (int j = i - 1; j > 0; --j) {
dp[j] = Math.min(dp[j - 1], dp[j]) + triangle.get(i).get(j);
}
dp[0] = dp[0] + triangle.get(i).get(0);
}
int minSum = dp[0];
for (int t : dp) {
minSum = Math.min(minSum, t);
}
return minSum;
}
}