力扣LeetCode: Python3解法 - 102.二叉树的层序遍历

力扣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/提供的思路和代码。