(Leetcode) 帕斯卡三角形 - Python实现

题目:

给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。Given a non-negative index k where k ≤ 33, return the kth index row of the Pascal's triangle.

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

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

--------------------------------------------------------------------------

解法1:杨辉三角生成器

class Solution(object):
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        # 创建生成器
        def yanghui_triangle():
            L = [1]
            while True:
                yield L
                L = [1] + [L[i-1] + L[i] for i in range(1,len(L))] + [1]

        # 输入为0,返回空列表
        if numRows == 0:
            return []
        n = 0
        li = []
        for i in yanghui_triangle():
            li.append(i)
            n += 1
            if n > numRows-1:
                break
        return li 

 

参考:

https://blog.csdn.net/poiuyds/article/details/87406578

你可能感兴趣的:(Leetcode)