要求
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
- 0 <= rowIndex <= 33
核心代码
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
i = 0
row = [1]
while i < rowIndex:
t = [x + y for x, y in zip(row,row[1:])]
row = [1, *t ,1]
i += 1
return row
解题思路:上面一道题是获取整个杨辉三角,在这道题中我们只需要获取杨辉三角的指定行即可,核心代码依旧不变t = [x + y for x, y in zip(row,row[1:])],只是最终的返回值变了而已,比较简单。