Python 杨辉三角生成器

最新推荐文章于 2023-03-10 16:11:33 发布

原创最新推荐文章于 2023-03-10 16:11:33 发布 · 344 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python学习随笔专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用Python生成器实现杨辉三角的算法。通过不断迭代和利用zip及sum函数，生成器能够高效地计算并输出杨辉三角的每一行，展示了生成器在节省内存和提高效率方面的优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding: utf-8 -*-
# 杨辉三角
"""
列表生成式直接占用空间，generator一边循环一边计算的机制，存储一个算法，可以通过for循环迭代调用generator
不能使用列表生成式可以通过函数实现，含有yield关键字则为生成器
普通函数返回一个结果，按照顺序执行
生成器函数返回的是一个生成器对象,每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行。
不停调用next非常繁琐并且没有更多元素时会报错，一次只返回一个对象，生成器正确用法是使用for循环迭代调用返回
"""


def triangels():
   ret = [1]
   while True:
      yield ret
      ret = [sum(i) for i in zip([0] + ret, ret + [0])]  
   #  ret = [x + y for (x, y) in zip([0] + ret,  ret + [0])]
   #  ret = [1] + [x+y for x,y in zip(ret[:-1],ret[1:])] + [1]
"""
上面的公式一开始不太理解，后面转化为下面的公式后比较一目了然，
zip([0]+ret,ret+[0)]=zip([0]+[1],[1]+[0])=zip[(0,1),(1,0)]
sum(i) 将两个列表相同位置的元素相加得到新的列表最后变成[1,1]
我们可以再看下当ret = [1,1]之后公式是怎么计算的,zip([0]+[1,1],[1,1]+[0])=zip[(0,1,1)+(1,1,0)]
sum(i) 或者x + y=[1,2,1]以此类推
"""
n = 0
results = []
for t in triangels():
   print('T = ', t)
   results.append(t)
   print('Results = ', results)
   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('测试失败！')