[]配列Leetcodeパスカルの三角形（118）

タイトル

numRowsの数ヤンホイトライアングルと世代前の非負整数numRowsの数、与えられました。

パスカルの三角形では、各番号は、その左上と右上の数です。

例：

输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

答え

class Solution:
    # Time: O(N^2)，Space: O(N^2)，N为numRows，题解也是O(N^2)
    def generate(self, numRows):
        if numRows == 0:
            return []
        if numRows == 1:
            return [[1]]
        if numRows == 2:
            return [[1], [1, 1]]
        a = [
            [1],
            [1, 1]
        ]
        for x in range(2, numRows):
            a.append([])
            a[x].append(1)
            for i in range(x-1):
                a[x].append(a[x-1][i]+a[x-1][i+1])
            a[x].append(1)
        return a


s = Solution()
print(s.generate(5))