【动态规划】Leetcode 118. 杨辉三角【简单】

杨辉三角

给定一个非负整数 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)。