LeetCode118-杨辉三角-Python

最新推荐文章于 2022-01-11 19:03:35 发布

wangjun0708

最新推荐文章于 2022-01-11 19:03:35 发布

阅读量252

点赞数

分类专栏：算法文章标签： LeetCode Python

本文链接：https://blog.csdn.net/weixin_42762089/article/details/87392302

版权

算法专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题目

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

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

示例:

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

思路

当 numRows=0,1,2 三种特殊情况时，返回固定结果。

当 numRows >2 时，循环调用 addToList()，将生成的子List ( 即代码中的 r[ ] ) 插入到母List ( 即代码中的 res[ ] ) 中

Python代码

class Solution:
    def generate(self, numRows: 'int') -> 'List[List[int]]':
        if numRows == 0:
            return []
        if numRows == 1:
            return [[1]]
        if numRows == 2:
            return [[1],[1,1]]

        res = [[1],[1,1]]
        i = 3
        while i <= numRows:
            self.addToList(res, i)
            i += 1
        return res
    
    def addToList(self, res, numRows):
        r = []
        for i in range(numRows):
            if i == 0 or i == numRows-1:
                r.append(1)
            else:
                r.append(res[-1][i-1]+res[-1][i])
        res.append(r)