欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python Solution LeetCode No.118 pascals-triangle(杨辉三角)

程序员文章站 2024-02-21 09:40:52
...

题目:

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

python Solution LeetCode No.118 pascals-triangle(杨辉三角)

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

示例:

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

代码:

class Solution:
    def generate(self, numRows):
        if numRows == 0:
            return []
        else:
            generate_list = []
            for i in range(numRows):
                level_generate_list = []
                for j in range(i+1):
                    if j == 0 or j == i:
                        level_generate_list.append(1)
                    else:
                        level_generate_list.append(generate_list[i-1][j-1] + generate_list[i-1][j])
                generate_list.append(level_generate_list)
        return generate_list

结果:

python Solution LeetCode No.118 pascals-triangle(杨辉三角)

说明: 

初始化空列表generate_list用于保存整个杨辉三角的二维列表,使用双重for循环,外层循环i用于控制,当前所在杨辉三角的层数,内层循环j用于控制杨辉三角当前层的所有元素。每次外层循环重新开始时将列表level_generate_list初始化为空,level_generate_list用来记录当前层的所有元素,由杨辉三角的特性可知,每层头尾两个元素都为1,所以j == 0 or j == i时,向level_generate_list中追加元素1。每层除头尾之外的中间元素的递推表达式为generate_list[i-1][j-1] + generate_list[i-1][j]。当前层循环完成后,将当前层列表追加到generate_list。最终两层循环都完成后,返回的generate_list即为解。