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

最新推荐文章于 2024-07-10 09:04:40 发布

原创最新推荐文章于 2024-07-10 09:04:40 发布 · 2.5k 阅读

3 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种使用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.youkuaiyun.com/poiuyds/article/details/87406578