Python生成器实现杨辉三角

在学习廖雪峰的python教程的时候,讲到生成器的时候,有一道练习是这样的:

练习
杨辉三角定义如下:
          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]
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('测试失败!')

这道题一开始我是用这样的方法解决的:

def triangles():
    L = [1]
    while True:  # 进入持续循环
        yield L # 输出数组
        L.append(0)  # 为数组添加最后一位
        L = [L[i - 1] + L[i] for i in range(len(L))]  

输出如下:
可以看到输出符合要求,但是显示测试失败。
没想到竟然测试失败,试下把断点打到main函数里面看看results里面是啥:
可以看到每一项的末尾都多出了一个0

为什么会多出一个零?

因为在triangles函数中,每次都会在L的末尾加上一个零,python中List是一个可变对象,因为每一次将当前List拷贝进去results的时候只是浅拷贝,可以理解成只拷贝了一个指针,因此加上一个零的操作会影响results中的List。

解决方法

直接贴代码了,其实就是加上了一个切片操作。

def triangles():
    L = [1]
    while True:  # 进入持续循环
        yield L[:len(L)] # 输出数组
        L.append(0)  # 为数组添加最后一位
        L = [L[i - 1] + L[i] for i in range(len(L))]  

测试通过

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
好的,您的问题是如何使用Python队列实现杨辉三角。 首先,让我们先定义一个生成杨辉三角的函数。 ```python def generate_pascal_triangle(num_rows): triangle = [] for row_num in range(num_rows): row = [None for _ in range(row_num + 1)] row[0], row[-1] = 1, 1 for j in range(1, len(row) - 1): row[j] = triangle[row_num - 1][j - 1] + triangle[row_num - 1][j] triangle.append(row) return triangle ``` 接下来,我们需要使用队列来实现这个函数。假设我们要生成杨辉三角的前n行。 首先,我们需要创建一个队列,将第一行的元素1推入队列中。 ```python from queue import Queue def generate_pascal_triangle(num_rows): triangle = [] q = Queue() q.put(1) ``` 接下来,我们开始遍历每一行。每当我们遍历一行时,我们就需要从队列中弹出一个元素,并将它添加到当前行的列表中。 ```python from queue import Queue def generate_pascal_triangle(num_rows): triangle = [] q = Queue() q.put(1) for row_num in range(num_rows): row = [] for i in range(row_num + 1): row.append(q.get()) triangle.append(row) ``` 当我们完成当前行的遍历时,我们需要将下一行的元素推入队列中。这可以通过在当前行的列表中添加1,然后遍历列表中除第一个和最后一个元素以外的其他元素来实现。 ```python from queue import Queue def generate_pascal_triangle(num_rows): triangle = [] q = Queue() q.put(1) for row_num in range(num_rows): row = [] for i in range(row_num + 1): row.append(q.get()) triangle.append(row) q.put(1) for j in range(len(row) - 1): q.put(row[j] + row[j+1]) q.put(1) return triangle ``` 最后,我们返回杨辉三角的列表即可。 这就是使用队列实现杨辉三角的方法。希望对您有所帮助!
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值