力扣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. 创建一个空的结果数组res和一个队列que,将根节点root放入队列que中。 2. 当队列que不为空时,执行以下操作: - 定义一个变量size,用于记录队列que的长度。 - 创建一个空的一维数组vec,用于记录当前层的节点值。 -0到size-1,执行以下操作: - 弹出队列que的头节点cur。 - 将cur的值cur->val添加到一维数组vec中。 - 若cur的左子节点不为空,则将cur的左子节点压入队列que中。 - 若cur的右子节点不为空,则将cur的右子节点压入队列que中。 - 将一维数组vec添加到结果数组res中。 3. 返回结果数组res作为层的结果。 以下是一段C++代码示例,实现了二叉树的层: ``` vector<vector<int>> levelOrder(TreeNode* root) { vector<vector<int>> res; // 结果数组 queue<TreeNode*> que; if (root) que.push(root); while (!que.empty()) { int size = que.size(); // 记录队列的长度 vector<int> vec; for (int i = 0; i < size; i++) { TreeNode* cur = que.front(); que.pop(); vec.push_back(cur->val); if (cur->left) que.push(cur->left); if (cur->right) que.push(cur->right); } res.push_back(vec); } return res; } ``` 这段代码使用了一个队列que,利用队列的先进先出的特点,按照层的顺将节点添加到结果数组res中。最后返回结果数组res即可得到二叉树的层结果。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [力扣刷题 | 二叉树](https://download.csdn.net/download/weixin_38633083/13740772)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [力扣:102. 二叉树的层](https://blog.csdn.net/qq_46111138/article/details/124895548)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值