Leetcode_Pascal's Triangle II

发表于 | 分类于

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

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

示例:

输入: 3
输出: [1,3,3,1]
进阶：

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

“””

假设当前层为[1,2,1]。 然后我们拷贝成两个复制副本a,b，

将0分别添加到a的开头和b的结尾;

那么我们有[0,1,2,1]和[1,2,1,0]。

然后，我们可以执行按对应元素相加操作，我们将有[1,3,3,1]。

这是来自帕斯卡三角形的定义。

“””

1
2
3
4
5
6
7
8
9
10
class Solution:
    def getRow(self, rowIndex):
        """
        :type rowIndex: int
        :rtype: List[int]
        """
        row = [1]
        for _ in range(rowIndex):
            row = [x + y for x, y in zip([0] + row, row + [0])]
        return row