119. 杨辉三角 II

119. 杨辉三角 II

给定一个非负索引 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]

你可能感兴趣的:(数据结构与算法,python)