目录
题目
119. 杨辉三角 II
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
0 <= rowIndex <= 33
解题思路
- 先把杨辉三角列出来,然后提取出那一行。
Code
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
if rowIndex == 0: return [1]
lis = [[1]]
while len(lis) <= rowIndex:
newRow = [i + j for i, j in zip(lis[-1] + [0], [0] + lis[-1])]
lis.append(newRow)
return lis[-1]