118. 杨辉三角

发布时间 2023-11-13 15:01:19作者: Frommoon

题目

  • 给定一个非负整数 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]]:
        triangle = []  # 用于存储杨辉三角形的二维列表
        for i in range(numRows):  # 循环生成每一行的元素
            row = [0] * (i+1)  # 初始化当前行的列表,长度为i+1
            row[0] = 1  # 当前行的第一个元素为1
            row[i] = 1  # 当前行的最后一个元素为1
            for j in range(1, i):  # 填充当前行的中间元素
                row[j] = triangle[i-1][j-1] + triangle[i-1][j]  # 根据上一行的元素计算当前行的元素
            triangle.append(row)  # 将当前行添加到杨辉三角形列表中
        return triangle  # 返回杨辉三角形的二维列表