leetcode 118.杨辉三角

117 阅读1分钟

118.杨辉三角

给定一个非负整数 *numRows,*生成杨辉三角的前 numRows 行。

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

输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]
class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        if numRows <= 1:
            return numRows * [[1]]

        prev = self.generate(numRows - 1)
        last = [sum(prev[-1][j-1:j+1]) for j in range(1, numRows-1)]
        prev.append([1, *last, 1])
        return prev
"""
首先判断特殊情况,numRows 为 0 则返回 [],为 1 返回 [[1]]
通过递归获得上一层结果 prev
通过上一层结果计算新的最后一层结果 last,
f(i,j) = f(i−1, j−1) + f(i−1, j) (两边的 1 会在下一行代码中另外加)
加入新行并返回
"""