用python生成器實現楊輝三角

先看楊輝三角的形態:函數

1
        1   1
      1   2   1
    1   3   3   1
  1   4   6   4   1
1   5   10  10  5   1


上學的時候大可能是用c語言的兩層for循環在實現,如今咱們嘗試用生成器來實現。

先說思路:咱們把每一次層看做一個list, 經過一個for循環,經過迭代,每次生成一個list,而生成器就在每一行生成list中起做用,咱們先定義一個函數,做用是生成每一行的list
1 for x in yhTriangle(10):
2     print(x)

 

對於每一行,list 的第一個元素和最後一個元素是不變的。若是用L = [] 表示的話, L[0], L[n],是 不變的,spa

[l[i] + l[i + 1] for i in range(len(l) - 1)]

因此用這個列表生成式表示 中間的那一部分code

1 def yhTriangle(n):
2     l, index = [1], 0
3     while index < n:
4         yield l
5         l = [1] + [l[i] + l[i + 1] for i in range(len(l) - 1)] + [1]
6         index += 1

完整代碼貼出來,你們看看,歡迎提意見啊哈blog

相關文章
相關標籤/搜索