Python中yield from使用的经典场景:二叉树的遍历

本文探讨了Python中`yield from`关键字在实现二叉树遍历的经典场景,通过示例展示了如何利用`yield from`进行先序、中序和后序遍历。

二叉树如下图所示:

Python中,可以使用队列数据结构来实现二叉树的层序遍历(也称为广度优先搜索)。这种遍历方式按照从上到下、从左到右的顺序逐层访问节点。下面是一个简单的例子,展示如何使用递归和迭代的方式来实现: **迭代法(利用deque)**: ```python from collections import deque class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def levelOrder(root): if not root: return [] queue = deque([root]) result = [] while queue: level_size = len(queue) current_level = [node.val for node in queue.popleft()] # 把当前层的所有节点添加到结果列表,并将他们的子节点加入队列 result.append(current_level) for _ in range(level_size - 1): queue.extend([node.left, node.right] if node else []) return result ``` **递归法**: ```python def levelOrderRecursion(root): if not root: return [] levels = [[root]] while levels: new_levels = [] for level in levels: for node in level: if node.left: new_levels.append(node.left) if node.right: new_levels.append(node.right) levels = new_levels yield from [node.val for level in levels for node in level] ``` 在这两个实现中,我们首先检查根节点是否为空,然后初始化队列并将根节点放入。接着在循环中,我们获取当前层的节点值并将其添加到结果,然后把它们的子节点依次加入队列。直到队列为空,说明已经遍历完所有层。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值