给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
解法一:BFS 广度优先遍历
#BFS思想:逐层遍历,每层的节点从左至右的顺序取出
def BFS(self, root: TreeNode):
if not root:
return None
ans = []
queue = [root] #从根节点开始,借助队列
while queue:
n = len(queue) #记录遍历当层的个数
tmp = [] #记录同一层的节点值
for _ in range(n):
q = queue.pop(0) #先进先出
if not q:
continue
tmp.append(q.val)
if q.left: #下层节点不为空时入队列,下次遍历时从队列里取
queue.append(q.left)
if q.right:
queue.append(q.right)
ans.append(tmp)#根据层级添加
return ans
借助层序遍历的思想,可以适用于镜像二叉树的解法
镜像二叉树:左右节点对称的二叉树
解法:BSF思想,逐层拿到的队列判断是回文(!!!注意叶子节点需补全,形成满二叉树)
class Solution:
def isSymmetric(self, root: TreeNode) -> bool:
if not root:
return True
queue = [root]
while queue:
n = len(queue)
tmp = []
for i in range(n):
q = queue.pop(0)
if not q: #为空补全叶子节点,关键点
tmp.append(-1)
continue
tmp.append(q.val)
queue.append(q.left)
queue.append(q.right)
if tmp[:] != tmp[::-1]: #判断列表是否回文
return False
return True