BM26 求二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路:
广度优先搜索。
定义一个队列。往队列中加入当前节点。当队列不为空的时候进行以下两个操作。
1. 求出当前队列的长度大小len
2. 取出队列前len个节点,每取出一个节点,就把对应节点的左右孩子入队(前提是左右孩子不为空)
然后重复上面的过程。
Python代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
result = []
if not root:
return result
queue = collections.deque()
queue.append(root)
while queue:
nowlevel = []
nowlength = len(queue)
for _ in range(nowlength):
node = queue.popleft()
nowlevel.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
result.append(nowlevel)
return result