杨辉三角
给定一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
解题思路
- 使用一个二维数组 dp 来存储杨辉三角的元素,然后将其转换为列表。
- 动态规划的递推关系式是 dp[i][j] = dp[i-1][j-1] + dp[i-1][j]。
Java实现
public class PascalTriangle {
public static List<List<Integer>> generate(int numRows) {
List<List<Integer>> triangle = new ArrayList<>();
// 处理 numRows 为 0 的情况
if (numRows == 0) {
return triangle;
}
// 初始化二维数组
int[][] dp = new int[numRows][numRows];
for (int i = 0; i < numRows; i++) {
dp[i][0] = 1;
}
// 填充二维数组
for (int i = 1; i < numRows; i++) {
for (int j = 1; j <= i; j++) {
//j <= i - 1 确定每行右边界j==i时补0。
// 因初始化int[] 默认值就是0,可以直接使用动态转移公式
// dp[i][j] = dp[i - 1][j - 1] + (j <= i - 1 ? dp[i - 1][j] : 0);
dp[i][j] = dp[i - 1][j - 1] + (dp[i - 1][j]);
}
}
// 将二维数组转换为列表
for (int i = 0; i < numRows; i++) {
List<Integer> row = new ArrayList<>();
for (int j = 0; j <= i; j++) {
row.add(dp[i][j]);
}
triangle.add(row);
}
return triangle;
}
public static void main(String[] args) {
int numRows = 5;
List<List<Integer>> result = generate(numRows);
// 打印结果
for (List<Integer> row : result) {
System.out.println(row);
}
}
}
时间空间复杂度
-
时间复杂度分析:需要填充二维数组 dp 的每个元素,总共需要填充约 1 + 2 + … + numRows = O(numRows^2) 个元素。
因此,时间复杂度为 O(numRows^2)。 -
空间复杂度分析:需要使用一个二维数组 dp 来存储杨辉三角的值,其大小为 numRows * numRows。因此,空间复杂度为 O(numRows^2)。