【题目】
Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.
For example, given the following triangle
[ [2], [3,4], [6,5,7], [4,1,8,3] ]
The minimum path sum from top to bottom is 11
(i.e., 2 + 3 + 5 + 1 =
11).
Note:
Bonus point if you are able to do this using only O(n) extra space, where n is the total number of rows in the triangle.
【Java代码】
public class Solution { /* 关键之处在于逆向思维。 * 根据题意会自然而然地想从上而下逐层寻找最优解,但是由于下层元素比上层多, * 边界处的计算非常繁琐。但是如果自下而上,逐层计算到当前层的最优解,那么 * 到达最顶端时,就是所求最优解。 */ public int minimumTotal(List<List<Integer>> triangle) { //先处理特殊情况 if (triangle == null || triangle.size() == 0) return 0; if (triangle.size() == 1) return triangle.get(0).get(0); int n = triangle.size(); int[] below = new int[n]; //用于保存下一层的最优解 int[] cur = new int[n]; //用于保存当前层的最优解 int i, j; //初始值为最下面一行的值 List<Integer> lastrow = triangle.get(n - 1); for (i = 0; i < n; i++) { below[i] = lastrow.get(i); } //从倒数第二行开始逐层向上计算 for (i = n - 2; i >= 0; i--) { List<Integer> row = triangle.get(i); //从底层到当前层每个位置的最优解取决于其下层临近的两个元素 for (j = 0; j < row.size(); j++) { if (below[j] < below[j + 1]) cur[j] = below[j] + row.get(j); else cur[j] = below[j + 1] + row.get(j); } //层次向上移动,当前层变为下层 for (j = 0; j < row.size(); j++) { below[j] = cur[j]; } } return cur[0]; } }
【扩展】
除了输出最小值之外,如何找出这条路径?
时间: 2024-10-23 00:11:46