当前位置: 首页 > news >正文

算法通过村第六关-树白银笔记|层次遍历

文章目录

  • 前言
  • 1. 层次遍历介绍
  • 2. 基本的层次遍历与变换
    • 2.1 二叉树的层次遍历
    • 2.2 层次遍历-自底向上
    • 2.3 二叉树的锯齿形层次遍历
    • 2.4 N叉树的层次遍历
  • 3. 几个处理每层元素的题目
    • 3.1 在每棵树行中找出最大值
    • 3.2 在每棵树行中找出平均值
    • 3.3 二叉树的右视图
    • 3.4 最底层最左边
  • 总结


前言


提示:这个世界上的任何一次拥抱都是将以松手告终。 --约瑟夫·布罗茨基《悲伤与理智》

二叉树的层次遍历是一个非常简单的问题,但是很多人没有接触过,所以就认为很难,面试的时候当然就写不出来了,今天我们就重点通关层次遍历这个问题。

1. 层次遍历介绍

广度优先在面试中属于常考类型,整体来说属于简单次,但是很多人面试遇到了就放弃,真的很可惜。我们就看看它的难度在哪里。

广度优先又叫做层次遍历,基本过程如下:
在这里插入图片描述
层次遍历就是从根节点开始,先访问根节点的一层全部元素,在访问之后的一层,类似金字塔一样一层一层访问。我们可以看做就是从左到右的一层一层去遍历二叉树,先访问5,之后访问3的左右孩子4 和 6 ,之后再访问4 和 6的孩子1 2 和 7 8。最后得到结果【5,4,6,1,2,7,8】

这里的问题就变成了将遍历过的子孩子保存下来,比如访问4 的左右孩子1 和 2 的时候应该先存储,直到处理完20之后再处理。使用队列的话就可以解决上述问题。

看下过程图解:

  1. 根节点入队 5
  2. 然后 5 出队, 之后将 5 的左右孩子4 6入队,保存在队列中
  3. 4 出队,将4 的左右孩子 1 2 入队
  4. 之后 6 出队,将 6 的左右孩子 7 8 入队,保存在队列中
  5. 最终 1 2 7 8 依次出队,此时都是叶子几点,只要出队就行了。

这个过程不复杂,如果将树的每层分开,是不是可以整活😎了?首先,能否将每层的元素顺序反转一下?或者说奇数不变,偶数反转?是否将输出层次从低到高(root)输出呢?再比如,既然可以拿到每层的元素,能不能找到当前层最大的元素?最小的元素呢?最右端的元素(右视图)?最左端的元素(左视图)?整个层的平均值?当然这都是可以实现的嘞😋,这么来回折腾有啥用呢?🤔,没啥用了,但是这些都是层次遍历的高频算法题目。

推荐力扣题目⭐⭐⭐⭐:

102. 二叉树的层序遍历 - 力扣(LeetCode)

107. 二叉树的层序遍历 II - 力扣(LeetCode)

199. 二叉树的右视图 - 力扣(LeetCode)

637. 二叉树的层平均值 - 力扣(LeetCode)

429. N 叉树的层序遍历 - 力扣(LeetCode)

515. 在每个树行中找最大值 - 力扣(LeetCode)

116. 填充每个节点的下一个右侧节点指针 - 力扣(LeetCode)

117. 填充每个节点的下一个右侧节点指针 II - 力扣(LeetCode)

103. 二叉树的锯齿形层序遍历 - 力扣(LeetCode)

2. 基本的层次遍历与变换

我们看一个最简单的情况,仅仅是遍历并输出全部元素,如下图:
在这里插入图片描述
上图的输入结果【5,4,6,7,8】,方法已经梳理过了,这里看一下代码怎么实现。先访问根节点,然后其左右孩子入队,接着出队,出队的元素将其左右孩子入队,直到队列为空就可以退出了:

	/*** 二叉树层序遍历* @param root* @return*/public static List<Integer> simpleLevelOrder(TreeNode root) {// 参数校验if (root == null) {return new ArrayList<Integer>();}// 创建存储空间List<Integer> res = new ArrayList<Integer>();LinkedList<TreeNode> queue = new LinkedList<TreeNode>();// 根节点先入队queue.add(root);// 遍历队列元素while(queue.size() > 0){// 获取当前队列的长度,这个长度相当于这一层的节点个数TreeNode t = queue.remove();res.add(t.val);if (t.left != null){queue.add(t.left);}if (t.right != null){queue.add(t.right);}}return  res;}

注意:

根据树的结构可以看出,一个节点在一层访问之后,其子孩子都是在下层按照FIFO的顺序处理的,因此队列在这里起到了一个缓存的作用。

如果要你将每层的元素分开该怎么做呢?看下下面的这个题:

2.1 二叉树的层次遍历

题目介绍:102. 二叉树的层序遍历 - 力扣(LeetCode)

在这里插入图片描述
在这里插入图片描述
我们再来看这个执行过程,我们先将根节点放入队列中,然后不断的遍历队列。
在这里插入图片描述
那如何判断某一层访问完了呢?这个简单,我们用一个size标记一下就行了,size表示某一层元素的个数,知道出队,size就减1,减到0就说明该层访问完了,当size变成0之后,这时队列中甚于元素个数恰好就是下一次的个数,因此重新将size标记为下一层的元素个数就可以继续处理新的一行,例如上面的序列中:

  1. 首先获取根节点 3 ,左右子节点不为空,将左右子节点放入队列,此时3 已经出队。剩下的9 20 恰好是第二层的所有节点,此时size = 2
  2. 继续将9 出队,左右节点不为空,将左右节点放入队列,size-- ,变成 1 ,之后将20 出队,左右节点不为空,将左右节点放入队列 size–, 变为 0。当size = 0,说明当前层已经处理完了,此时队列有4个元素,恰好是下一层的元素个数。
  3. 最后,我们把每层遍历到的节点放入一个结果集中,将其返回就可以了

按层次遍历经典代码:

	/*** 层次遍历  经典方式* @param root* @return*/public static List<List<Integer>> level102Order(TreeNode root) {// 校验参数if (root == null){return new ArrayList<List<Integer>>();}// 创建空间List<List<Integer>> res = new ArrayList<List<Integer>>();LinkedList<TreeNode> queue = new LinkedList<TreeNode>();// 将根节点放入队列  然后不断遍历队列queue.add(root);while(queue.size() > 0){// 获取当前size的个数  即当前层有多少个元素int size = queue.size();// 创建临时空间  将一层的元素存入其中List<Integer> temp = new ArrayList<Integer>();for(int i = 0; i < size; i++){TreeNode t = queue.remove();temp.add(t.val);if (t.left != null){queue.add(t.left);}if (t.right != null){queue.add(t.right);}}// 此时的temp就是一层的所有元素,用list类型保存temp,最后加入到结果集中res.add(temp);}return res;}

此方法一定要牢记,在算法体系中是很重要的一环,他与链表反转,二分查找可以相提并论,务必掌握,彻底理解,然后记忆。

当然上面的算法理解了,下面的一系列题目都变得简单了,不信你往下看。

注意:

在Java中队列的实现不止一种,需要注意。

2.2 层次遍历-自底向上

题目介绍:107. 二叉树的层序遍历 II - 力扣(LeetCode)
在这里插入图片描述
在这里插入图片描述
如果要求从上到下输出每一层的节点值,做法很直观,在遍历完一层之后,将该层存储节点值的列表添加到结果集的尾部。这道题要求从下到上输出每一层的节点值,只要对上述操作稍作修改就可以了,在遍历完一层节点后,将存储该层节点值的列表添加到结果集的头部。

为了降低在结果列表的头部添加一层节点值的列表的时间复杂度,结果列表可以使用链表的结构,在链表头部添加一层节点值的列表的时间复杂度为O(1),在Java中,由于我们需要返回一个List接口,在这里可以使用链表实现。

代码就很普通了😎:

	/*** 层序遍历  自底向上* @param root* @return*/public static List<List<Integer>> levelOrderBottom(TreeNode root) {// 校验参数if (root == null){return new ArrayList<List<Integer>>();}// 创建空间List<List<Integer>> levelOrder = new LinkedList<List<Integer>>();Queue<TreeNode> queue = new LinkedList<TreeNode>();// 根节点存入队列中  不断的遍历queue.offer(root);while(queue.size() > 0){List<Integer> level = new ArrayList<Integer>();int size = queue.size();for(int i = 0; i < size; i++){TreeNode node = queue.poll();level.add(node.val);// 这样也行TreeNode left = node.left,right = node.right;if(left != null){queue.offer(left);}if (right != null){queue.offer(right);}}levelOrder.add(0,level); // 放在头部 栈}return levelOrder;}

2.3 二叉树的锯齿形层次遍历

题目介绍:103. 二叉树的锯齿形层序遍历 - 力扣(LeetCode)
在这里插入图片描述
在这里插入图片描述
这个题也是102. 二叉树的层序遍历 - 力扣(LeetCode)的变种,只是最后输出的要求变化了,要求我们按照层数的奇偶来决定每层输出的顺序,如果当前是偶数层,从左到右输出当前层的节点值,否则,从右至左输出当前层的节点值。这里我们同样可以采用102的方法思想。

但是为了满足条件,【左右->右左】交替暑促的锯齿形,可以利用【双端队列】的数据机构来维护当前节点值的输出顺序。双端队列是一个可以在队列任意一端插入元素的队列。在广度优先遍历搜索当前节点,扩展下一层节点的时候我们仍然可以从左往右循序扩展,但是对于当前层节点的存储我们维护一个变量isOrderLeft记录时从左到右还是从右到左:

  • 如果从左到右,我们每次将被遍历的元素插入到双端队列的末尾
  • 从右到左,我们每次将被遍历的元素插入到双端队列的头部。

代码展示😋

/***  锯齿层次遍历* @param root* @return*/public static List<List<Integer>> zigzagLevelOrder(TreeNode root) {// 校验参数if (root == null){return new ArrayList<List<Integer>>();}// 创建空间List<List<Integer>> ans = new LinkedList<List<Integer>>();Queue<TreeNode> queue = new LinkedList<TreeNode>();// 将根节点放入队列  不断循环遍历queue.offer(root);// 判断奇偶变量boolean isOrderLeft = true;while(queue.size() >0){// 这里要使用双端队列Deque<Integer> temp = new LinkedList<Integer>();int size = queue.size();for(int i = 0; i < size; i++){TreeNode t = queue.poll();if (isOrderLeft){temp.offerLast(t.val);}else{temp.offerFirst(t.val);}if (t.left != null){queue.offer(t.left);}if (t.right != null){queue.offer(t.right);}}ans.add(new LinkedList<Integer>(temp));isOrderLeft = !isOrderLeft;}return ans;}

2.4 N叉树的层次遍历

题目介绍:429. N 叉树的层序遍历 - 力扣(LeetCode)

在这里插入图片描述
在这里插入图片描述
首先看下N叉树的结构,就是一个值的不同,变成了以一个列表,其类型仍然时Node:

class TreeNode {public int val;public List<TreeNode> children;}

也就是说我们可以说,这个也是102. 二叉树的层序遍历 - 力扣(LeetCode)的变种,我们就用简单的广度优先,借助队列即可实现。

	/*** N 叉树的遍历* @param root* @return*/public static List<List<Integer>> nLevelOrder(NTreeNode root) {// 校验参数if (root == null){return new ArrayList<List<Integer>>();}// 创建空间List<List<Integer>> res = new ArrayList<List<Integer>>();Deque<NTreeNode> queue = new ArrayDeque<NTreeNode>();// 根节点放入队列 不断遍历queue.offer(root);while(!queue.isEmpty()){Deque<NTreeNode> next = new ArrayDeque<>();List<Integer> level = new ArrayList<>();while(!queue.isEmpty()){NTreeNode current = queue.pollFirst();level.add(current.val);// 注意这里链表遍历for (NTreeNode child : current.children){if (child != null){next.add(child);}}}// 重新定义queuequeue = next;res.add(level);}return res;}

3. 几个处理每层元素的题目

如果我们拿到每一层的元素,那么是不是可以利用一下就可以造几个题目出来呢?例如每层找最大的值,平均值,最右侧的值?当然可以。这也是力扣的上面比较经典的题目了:

题目推荐⭐⭐⭐⭐:

515. 在每个树行中找最大值 - 力扣(LeetCode)

637. 二叉树的层平均值 - 力扣(LeetCode)

199. 二叉树的右视图 - 力扣(LeetCode)

当然我们自己也可以造题目,比如说层次取最小值可以吗?每层的最左侧值可以不?是不是现在发现算法其实很好玩的🥰

3.1 在每棵树行中找出最大值

题目介绍:515. 在每个树行中找最大值 - 力扣(LeetCode)

在这里插入图片描述
这里其实就是得到一层之后使用一个变量来记录当前的最大值:

代码也很简单:

	/*** 层序遍历找到每层最大值* @param root* @return*/public static List<Integer> largestValues(TreeNode root) {// 校验参数if(root == null){return new ArrayList<Integer>();}// 创建空间List<Integer> res = new ArrayList<Integer>();Deque<TreeNode> queue = new ArrayDeque<TreeNode>();// 根节点入队,不断循环遍历queue.offer(root);while(queue.size() > 0){int size = queue.size();int maxLevel = Integer.MIN_VALUE;for(int i = 0; i < size; i++){TreeNode t = queue.poll();maxLevel = Math.max(maxLevel, t.val);if (t.left != null){queue.offer(t.left);}if (t.right != null){queue.offer(t.right);}}res.add(maxLevel);}return res;}

3.2 在每棵树行中找出平均值

题目介绍:637. 二叉树的层平均值 - 力扣(LeetCode)

在这里插入图片描述
在这里插入图片描述
这个题目和前面的一样,之不多每层都先将元素保存下来,最后求一个平均值就可以了,是不是是很简单;

	/*** 层序遍历,求出每层平均值* @param root* @return*/public static List<Double> averageOfLevels(TreeNode root) {// 检验参数if(root == null){return new ArrayList<Double>();}// 创建空间List<Double> res = new ArrayList<Double>();Queue<TreeNode> queue = new LinkedList<TreeNode>();// 根节点入队,不断循环queue.offer(root);while(queue.size() > 0){int size = queue.size();double sum = 0;for(int i = 0; i < size; i++ ){TreeNode t = queue.poll();sum += t.val;if (t.left != null){queue.offer(t.left);}if (t.right != null){queue.offer(t.right);}}res.add(sum/size);}return res;}

3.3 二叉树的右视图

题目介绍:199. 二叉树的右视图 - 力扣(LeetCode)
在这里插入图片描述
在这里插入图片描述
这个题目说实话出频率还挺高的,如果没有提前思考过,很难想到答案。其实也简单,在BFS进行层次遍历的时候,记录下每层的最后一个元素就可以了。

   /*** 层序遍历,输出右视图* @param root* @return*/public static List<Integer> rightSideView(TreeNode root) {// 校验参数if(root == null){return new ArrayList<Integer>();}// 创建空间List<Integer> res = new ArrayList<Integer>();Queue<TreeNode> queue = new LinkedList<TreeNode>();// 根节点入队,不断循环遍历queue.offer(root);while(!queue.isEmpty()){int size = queue.size();for(int i = 0; i < size; i++){TreeNode t = queue.poll();if (t.left != null){queue.offer(t.left);}if (t.right != null){queue.offer(t.right);}// 这里将最后一个节点放入结果集中if (i == size - 1){res.add(t.val);}}}return res;}

是不是有点成就感了,感觉算法是有套路的,并不是四刷题目。这三道题基本上都是层序遍历的变种。

那我我们也可以尝试着造出题目来:

有右视图,那么左视图行不行?俯视图行不行?想一下💡

嘿嘿😁,我这里告诉你吧,俯视图不行,但是左视图可以,想下为什么?🤔

3.4 最底层最左边

题目介绍:513. 找树左下角的值 - 力扣(LeetCode)

在这里插入图片描述
在这里插入图片描述
在第二部分,我们写了很多层次遍历的变种形式,这里有两个问题:

  1. 该怎么知道什么时候到最底层了?
  2. 加入最底层有两个,该怎么知道那个时最左的呢?

带着这两个问题,我们再看下层序遍历的执行过程:
在这里插入图片描述
我们可以发现,正常执行层次遍历的,不管最底层有几个元素,最后输出的一定是底层最右的元素7,这里我们就想象一下,可不可以和上面题目一样结合反转操作实现呢,每一层先反转再存入队列,到最后是不是输出的为最左节点值呢?是的,知道了这个解题关键点,我想答案你已经会写了🥰。

	public int findBottomLeftValue(TreeNode root) {// 校验参数if(root.left == null && root.right == null){return root.val;}// 创建空间Queue<TreeNode> queue = new LinkedList<TreeNode>();// 根节点入队,不断循环遍历queue.offer(root);TreeNode temp = new TreeNode();while(!queue.isEmpty()){temp = queue.poll();// 先右后左	if(temp.right != null){queue.offer(temp.right);}if(temp.left != null){queue.offer(temp.left);}}return temp.val;}

总结

提示:二叉树的层序遍历,各种变种问题,熟练掌握模板题,总结套路,是不是感觉算法也是有迹可循的😎

http://www.lryc.cn/news/164311.html

相关文章:

  • SpringCloud理解篇
  • 编写LED灯的驱动,实现三盏灯的控制
  • Flink报错处理-1
  • bim与数字孪生智能建造的关系
  • 【Linux】进程篇(补):守护进程
  • SpringMVC自定义视图完成步骤 和 视图解析的源码剖析
  • 合宙Air724UG LuatOS-Air lvgl字库
  • C#,数值计算——指数微分(exponential deviates)的计算方法与源程序
  • ADAS自动驾驶
  • Python从零到一构建项目
  • 使用todesk或者向日葵远程Ubuntu22.04系统的客户机黑屏
  • JBoss JMXInvokerServlet 反序列化漏洞复现(CVE-2015-7501)
  • 比Mojo慢68000倍,Python性能差的锅该给GIL吗?
  • CSS读书笔记
  • Qt使用QSqlDatabase remoeDatabase()连接提示仍在使用解决方案
  • 管易云与金蝶云星空对接集成仓库查询打通仓库新增
  • ubuntu 安装 Mongodb 4.0、4.2、4.4
  • 详解Hugging Face Transformers的TrainingArguments
  • 【LeetCode-中等题】17. 电话号码的字母组合
  • 读高性能MySQL(第4版)笔记06_优化数据类型(上)
  • mac如何创建mysql数据库
  • Ceph入门到精通-centos8 install brctl
  • sqli第一关
  • 入行IC | 新人入行IC选择哪个岗位更好?
  • 时间旅行的Bug 奇怪的输入Bug
  • 解决nbsp;不生效的问题
  • 【Lidar】Cloud Compare介绍安装包
  • Java中的Maven是什么?
  • 计算机操作系统
  • 海学会读《乡村振兴战略下传统村落文化旅游设计》2023年度许少辉八一新书