给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1] 输出:[[1]]
示例 3:
输入:root = [] 输出:[]
思路:实现二叉树的层序遍历,顾名思义,从左到右遍历每一层出现的节点,实际上也就是先遍历根节点,然后遍历根节点的左子树,然后右子树,若左子树还有左右子树,先遍历他的左子树,然后后子树,直到结束。很明显是一种先进先出的结构,故我们采用队列这种数据结构来实现。deque是一个双端队列,顾名思义左右两端都能添加和删除元素,因此我们可以使用右端进左端出来实现先入先出,根节点先入队列,然后弹出时,判断是否有左右子树,若有先加入左子树后加入右子树,最后依次弹出即可。
代码(Python):
class Solution(object):
def levelOrder(self, root):
if not root: #判空
return []
result = []
deque = collections.deque([root]) #利用双端队列,初始将root放入
while deque:
level = [] #每一层的节点
for _ in range(len(deque)):
cur = deque.popleft() #先入先出,左端弹出先进入的节点
level.append(cur.val)
if cur.left:
deque.append(cur.left)
if cur.right:
deque.append(cur.right)
result.append(level)
return result