練習
楊輝三角定義如下:
1
/ \
1 1
/ \ / \
1 2 1
/ \ / \ / \
1 3 3 1
/ \ / \ / \ / \
1 4 6 4 1
/ \ / \ / \ / \ / \
1 5 10 10 5 1
把每一行看做一個list,試寫一個generator,不斷輸出下一行的list:
期待輸出: # [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]
思路:找規律
第一行:[1]
第二行:[0, 1] + [1, 0] = [1, 1]
第三行:[0, 1, 1] +?[1, 1, 0] = [1, 2, 1]
第四行:[0, 1, 2, 1] +?[1, 2, 1, 0] = [1, 3, 3, 1]
第五行:[0, 1, 3, 3, 1] +?[1, 3, 3, 1, 0] = [1, 4, 6, 4, 1]
…… ……
可以看出,每一行的List都是由上一行的List分別在頭、尾加上一個0之后新生成的兩個List對應位置元素之和,求兩個List對應位置元素之和可以利用 zip()函數
即:L=[ sum[ i ] for i in zip( [0]+L, L+[0] ) ]
?
|
? 代碼:
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import math
def triangles():
L = [1]
while True:
yield L
zipped=list(zip([0]+L, L+[0]))
L = [sum(i) for i in zip([0]+L, L+[0])]
# 期待輸出:
# [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]
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('測試失敗!')
?
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
