(第二天)
题目:
给定一个非负整numRows(1 <= numRows <= 30),生成「杨辉三角」的第numRows行。 在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例1
输入: numRows = 1 输出: [[1]]
示例2
输入: numRows = 5 输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
题目分析,杨辉三角的两个腰都是一,说明列表的第一个和最后一个都是1,除了首尾两个数字(因为首位两数在杨辉三角的边界上面没有两个相邻的数),中间的数字都是上一个列表的相邻两数的和(n-1)+(n)。传入的参数为numRows即为杨辉三角形的行数,最长列表包含的个数。
def generate(numRows):
lis = [] # 创建空列表
for i in range(numRows): # 利用range生成从0到numRows的数
li = [1]*(i+1) # 利用for循环不断生成列表,列表个数于numRows个数相同
输出如下
这时列表里的元素都为1,需要把中间的1替换成正确的数,替换规则便是用上一个列表相邻的两个数相加。
def generate(numRows):
lis = []
for i in range(numRows):
li = [1]*(i+1)
if i >= 2: # 当i大于2即numbRows>3,行数大于3,列表中间的数需要替换
for j in range(1, i): # 从1开始生成因为需要替换的数字不包含首位的1
li[j] = li_0[j-1] + li_0[j] # li_0为上一个列表
lis += [li] # 将子列表添加到要输出的列表
li_0 = li # 这一语句是在第二个for循环外,所以当numRows为2时,li_0 = [[1,1]]
for item in lis: # 从主列表一个一个输出列表达到阶梯型效果
print(item)
generate(6)
输出结果如下
可以在改进一下,可以随意输入行数numRows
def generate(numRows):
lis = []
for i in range(numRows):
li = [1]*(i+1)
if i >= 2:
for j in range(1, i):
li[j] = li_0[j-1] + li_0[j]
lis += [li]
li_0 = li
return lis
if __name__ == '__main__':
while True: # 使用循环,可以多次输入
numRows = int(input("enter numRows:")) # 注意!input获得的是字符串,需要用int转换成数值
lis = generate(numRows)
for item in lis:
print(item)
还要注意的是第二个for循环的位置,不要放在def generat函数中,如果放在def genera函数中则需要等待所有子列表添加完,不然for获取到第一个子列表就会输出,向下面这样
来源参考:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/pascals-triangle



