杨辉三角 II

题目描述

来源于 https://leetcode-cn.com/

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

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

示例:

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

进阶:

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

解法:

中规中矩的实现,不知道第 n 行的值,是不是有一个推导公式可以直接得出来,不过不想深究了。

class Solution:
    def getRow(self, row_index: int) -> List[int]:
        row = [1] * (row_index + 1)
        
        for i_row in range(row_index):
            for i in range(i_row, 0, -1):
                row[i] = row[i-1] + row[i]

        return row