题目
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
思路
这是我在力扣上看见的一个巧解的做法,在上一行的首尾都加上0之后相加,例如:121与1331,121变为0121和1210,两个相加之后1331
代码
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
res=[]
if numRows==0:
return res
res=[[1]]
while len(res)<numRows:
num=[a+b for a,b in zip([0]+res[-1],res[-1]+[0])]
res.append(num)
return res