LeetCode第118题：杨辉三角

题目

给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中，每个数是它左上方和右上方的数的和。

• 示例 1:
  输入: numRows = 5
  输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

• 示例 2:
  输入: numRows = 1
  输出: [[1]]

代码

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        if numRows == 1:
            return [[1]]
        res = [[1]]
        for i in range(1, numRows):
            result = []
            for j in range(i + 1):
                if j == 0:
                    result.append(1)
                elif j == len(res[i-1]):
                    result.append(1)
                else:
                    result.append(res[i-1][j] + res[i-1][j - 1])
            res.append(result)
        return res

解释

• 在函数中，首先判断如果 numRows 等于 1，则直接返回 [[1]]，因为只有一行的时候直接返回 [1]。

• 接着，初始化 res = [[1]]，表示第一行为 [1]。

• 然后，从第二行开始，循环 numRows-1 次，表示共需要计算 numRows-1 行。在每次循环中，定义一个空列表 result 来存储当前行的结果。接着，遍历当前行的每一个元素，根据杨辉三角的规律，如果是当前行的第一个元素或最后一个元素，则直接添加 1 到 result 中，否则将当前位置的值设置为上一行的 i-1 位置和 i 位置的值之和。最后将 result 添加到 res 中。

• 循环结束后，返回 res，即杨辉三角的前 numRows 行。

• 总体来说，这是一个比较简单的实现。时间复杂度为 O(numRows^2)，空间复杂度为 O(numRows^2)。