118 杨辉三角
给定一个非负整数 *numRows
,*生成「杨辉三角」的前 numRows
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
先把numRows等于1和2的情况单独写出来,后面再依次按照索引去找上一个数组中 对应的数加起来,注意if if else会报错,因为后面的if else会成为一组,导致i==0的结果[1]重复添加进List,if elif else正确,表示当条件 1 和 条件 2 都不为 true 时执行的代码。
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
List = []
for i in range(numRows):
if i == 0:
List.append([1])
elif i == 1:
List.append([1,1])
else:
nums = [0] * (i+1)
nums[0], nums[-1] = 1, 1
for j in range(1,i):
nums[j] = List[i-1][j-1] + List[i-1][j]
List.append(nums)
return List