题目:
杨辉三角定义如下:
1
/ \
1 1
/ \ / \
1 2 1
/ \ / \ / \
1 3 3 1
/ \ / \ / \ / \
1 4 6 4 1
/ \ / \ / \ / \ / \
1 5 10 10 5 1
把每一行看做一个list,试写一个generator,不断输出下一行的list:
解题思路见源码
# -*- coding: utf-8 -*-
# 主要部分-generator生成器的编写
def triangles():
L=[1]
while True:
yield L
L=[L[i-1]+L[i] for i in range(len(L)) if i!=0]#将下标不为0的部分赋值
L.insert(0,1)#上一行执行完得到的L长度为上一个L的长度-1,缺少下标为0和len(L*)+1的部分
L.append(1)#在缺的位置各添一个1即可
#主函数部分—廖老师的源码
def main():
n = 0
results = []
for t in triangles():
print(t)
results.append(t)
n = n + 1
if n == 10:
break
if results == [
[1],
[1, 1],
[1, 2, 1],
[1, 3, 3, 1],
[1, 4, 6, 4, 1],
[1, 5, 10, 10, 5, 1],
[1, 6, 15, 20, 15, 6, 1],
[1, 7, 21, 35, 35, 21, 7, 1],
[1, 8, 28, 56, 70, 56, 28, 8, 1],
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
]:
print('测试通过!')
else:
print('测试失败!')
print(results)
main()