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]]:
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 # 返回杨辉三角形的二维列表