给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
0 <= rowIndex <= 33
来源:力扣(LeetCode)
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
if rowIndex == 0:
return [1]
elif rowIndex == 1:
return [1,1]
else:
res = [[1],[1,1]]
for i in range(2,rowIndex+1):
list1 = res[-1]
list2 = [1]
for j,k in zip(list1[:-1],list1[1:]):
sum = j + k
list2.append(sum)
list2.append(1)
res.append(list2)
return res[-1]