python Solution LeetCode No.118 pascals-triangle(杨辉三角)
程序员文章站
2024-02-21 09:40:52
...
题目:
给定一个非负整数 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:
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
结果:
说明:
初始化空列表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即为解。
上一篇: 在OC的基础之上学习Swift:常量变量及类型推导
下一篇: vim的简单配置