python二叉树层序遍历代码
时间: 2024-03-16 19:40:25 浏览: 85
python完全二叉树的层序遍历
当然可以!下面是一个Python的二叉树层序遍历的代码示例:
```python
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 []
result = []
queue = [root]
while queue:
level = []
size = len(queue)
for _ in range(size):
node = queue.pop(0)
level.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
result.append(level)
return result
```
这段代码定义了一个`TreeNode`类来表示二叉树的节点,其中包含了节点的值、左子节点和右子节点。`levelOrder`函数实现了二叉树的层序遍历,它使用了一个队列来辅助遍历过程。首先将根节点加入队列,然后在每一层遍历时,将当前层的节点值添加到结果列表中,并将下一层的节点加入队列。最后返回结果列表即可。
阅读全文