> 文档中心 > Python每日一练

Python每日一练

(第二天)

题目:

给定一个非负整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 lisif __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