杨辉三角python_118. 杨辉三角（Python）

题目

难度：★☆☆☆☆

类型：数组

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

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

示例

输入: 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: # 输入为0，输出[]

return []

if numRows == 1: # 输入为1，输出第一层

return [[1]]

if numRows == 2: # 输入为2，输出前两层

return [[1], [1, 1]]

res, prev_layer = [[1], [1, 1]], [1, 1] # 初始化结果变量和上一层结果

for r in range(2, numRows):

cur_layer = [] # 初始化

for i in range(0, len(prev_layer)-1):

cur_layer.append(prev_layer[i]+prev_layer[i+1]) # 当前层结果

cur_layer = [1] + cur_layer + [1] # 左右两边添加1

res.append(cur_layer) # 添加当前层结果

prev_layer = cur_layer # 更新前一层结果为当前层结果

return res

如有疑问或建议，欢迎评论区留言~