力扣LeetCode: Python3解法 - 102.二叉树的层序遍历
一、BFS和DFS是什么?
二叉树由于其本身特殊的树状结构,常采用BFS(宽度优先搜索)和DFS(深度优先搜索)两种方式进行求解。两种搜索过程的对比如下:
(1)BFS:指依次遍历二叉树的每层节点,从左至右添加完后再往下一层走,与层序遍历的顺序很吻合;
(2)DFS:先从上至下添加完后,再往右边走,具体遍历顺序见上图。
可直接采用BFS思想实现层序遍历,也可采用DFS结合递归思想实现层序遍历
二、BFS实现层序遍历
思路:使用队列存储当前层所有没有被遍历过的节点,再依次弹出,添加其val值到目标数组中,迭代完每层后再输出最终结果。
1.数据初始化
deque = collections.deque() # 定义双边队列
deque.append(root) # 添加第一层:根节点
res = [] # 存储最终的层序遍历结果
2.迭代遍历
while deque: # 当没有节点时,deque为空,自动跳出循环
n = len(deque) # 该层的节点个数
tmp = [] # 初始化每层的遍历结果
for _ in range(n): # 对于该层的所有节点
cur = deque.popleft() # 依次弹出
if not cur:
continue # 可能搜寻到null,忽略
tmp.append(cur.val) # 添加当前节点的值
deque.append(cur.left) # 并添加当前节点的左右子节点
deque.append(cur.right) # 等待下一层的遍历
if tmp:
res.append(tmp) # 添加每层的结果
return res
三、DFS实现层序遍历
思路:使用递归思想(根左右)依次遍历二叉树,并设置一个行为level的数组存储每层的数据。
1.递归函数
def dfs(root, level, res):
if not root: return
if len(res) == level: res.append([]) # 为第level层新建一行
res[level].append(root.val) # 将当前根节点赋给第level层
if root.left: dfs(root.left, level+1, res) # 当在同一层就不会新建
if root.right: dfs(root,right, level+1, res)
2.主函数
res = []
dfs(root, 0, res)
return res
总结
以上就是今天要讲的内容,谢谢力扣用户@负雪明烛https://leetcode.cn/u/fuxuemingzhu/提供的思路和代码。