leetcode_119 杨辉三角 II

57 阅读1分钟

要求

给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。

在「杨辉三角」中,每个数是它左上方和右上方的数的和。

Animated2.gif

示例 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

image.png

解题思路:上面一道题是获取整个杨辉三角,在这道题中我们只需要获取杨辉三角的指定行即可,核心代码依旧不变t = [x + y for x, y in zip(row,row[1:])],只是最终的返回值变了而已,比较简单。