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