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

  • 5
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值