119. 杨辉三角 II

简单

相关标签

premium lock icon

相关企业

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

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

示例 1:

输入: rowIndex = 3

输出: [1,3,3,1]

示例 2:

输入: rowIndex = 0

输出: [1]

示例 3:

输入: rowIndex = 1

输出: [1,1]

提示:

0 <= rowIndex <= 33

进阶：

你可以优化你的算法到 O(rowIndex) 空间复杂度吗？

class Solution:

    def generate(self, numRows: int) -> List[List[int]]:
        if numRows == 1:
            return [[1]]
        elif numRows == 2:
            return [[1],[1,1]]
        else:
            last = self.generate(numRows-1)

            lis = last[-1]
            l,r = 0,1
            res = [1]
            while r<len(lis):
                res.append(lis[l]+lis[r])
                l+=1
                r+=1
            res+=[1]

            last.append(res)

            return last


    def getRow(self, rowIndex: int) -> List[int]:
        res = self.generate(rowIndex+1)
        return res[-1]

这题没有人做吗？这么简单的思路竟然又是双百