Leetcode 118 杨辉三角

Posted 2020-12-25 Aprilnn

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

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

示例:

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

• 解答

class Solution:
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        if numRows==0:
            return []
        if numRows==1:
            return [[1]]
        if numRows==2:
            return [[1],[1,1]]
        l1=[1]
        l2=[1,1]
        l=[l1,l2]
        i=2
        while i<numRows:
            l3 = []
        #创建一个空列表来存储下一行的数据，每一次循环都清空列表
            l3.append(1)#第一个和最后一个元素固定为1
            for j in range(len(l2)-1):
                l3.append(l2[j]+l2[j+1])
            l3.append(1)
            l2=l3
            l.append(l3)#将列表添加到要返回的列表中
            i+=1
        return l