leetcode 刷题 119. 杨辉三角II

111 阅读1分钟

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

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

示例:

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

解答:

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        rownum = rowIndex + 1
        result = [ [1] * (i+1) for i in range(rownum)]
        if rownum>=3:
            for i in range(2,rownum):
                for j in range(1,i):
                    result[i][j] = result[i-1][j-1] + result[i-1][j]
        output = result[rowIndex]
        return output