创建一个二叉树并实现二叉树的层序遍历
时间: 2023-06-18 12:06:53 浏览: 59
下面是 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: TreeNode) -> List[List[int]]:
if not root:
return []
res = []
queue = [root]
while queue:
level = []
for i in range(len(queue)):
node = queue.pop(0)
level.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(level)
return res
# 创建一个二叉树,如下所示
# 1
# / \
# 2 3
# / \ \
# 4 5 6
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.right = TreeNode(6)
# 输出二叉树的层序遍历结果
print(levelOrder(root)) # [[1], [2, 3], [4, 5, 6]]
```
以上代码中,我们先定义了二叉树节点类 `TreeNode`,然后实现了二叉树的层序遍历函数 `levelOrder`,最后利用这个函数创建了一个二叉树并输出了它的层序遍历结果。