题目地址:
https://leetcode.com/problems/triangle/
给定一个三角形的数组,问从最上一层出发到最下一层中所有路径和最小的那个和是多少。每次只能从某一层向下走,并且只能走相邻的方格。经典的动态规划问题。我们可以设一个二维数组,每个位置表示,从 ( 0 , 0 ) (0,0) (0,0)位置出发到当前位置的所有路径的最小和。那么这个路径只有两种可能性,我们只需要在这两种可能性里取较小的即可。最后遍历完三角形后返回最后一层最小值即可。为了节省空间,我们可以只用一维数组进行更新。但需要注意的是,由于每一层更新的时候需要用到之前一层的信息,所以更新数组的时候必须从右向左更新。代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
public int minimumTotal(List<List<Integer>> triangle) {
// 开一个长度为三角形最后一行size的数组,并初始化为无穷大
int[] res = new int[triangle.get(triangle.size() - 1).size()];
Arrays.fill(res, Integer.MAX_VALUE);
// 先初始化第0个数字
res[0] = triangle.get(0).get(0);
// 接着开始对数组进行更新,从上向下遍历三角形,并从右向左更新数组
for (int i = 1; i < triangle.size(); i++) {
for (int j = triangle.get(i).size() - 1; j > 0; j--) {
// 取上一层中较小的邻接点进行更新
res[j] = Math.min(res[j], res[j - 1]) + triangle.get(i).get(j);
}
// 最后更新第0位,因为第0位只有一个邻接点,如果在上面更新会造成数组越界
res[0] += triangle.get(i).get(0);
}
// 最后找一下最小值
int ans = Integer.MAX_VALUE;
for (int i = 0; i < res.length; i++) {
ans = Math.min(ans, res[i]);
}
return ans;
}
}
时间复杂度 O ( n 2 ) O(n^2) O(n2)( n n n为三角形的边长),空间 O ( n ) O(n) O(n)。