稀土掘金首页沸点课程直播活动竞赛商城APP邀请有礼插件探索稀土掘金搜索创作者中心vip会员4Java查序的头像一篇文章带你玩转二叉树的层序遍历 | 十道题巩固练习

题目描述

解题思路

• 由题可知，要求输出是按照二叉树每层的元素来做输出

• 我使用队列来对二叉树每层元素进行存储和输出

• 根据队列的长度可以判断出当前层的元素个数并遍历

• 首先去判断入参是否为 null

  • 如果为 null则直接返回 null

• 创建队列对象

• 将 root放入队列

• 遍历队列是否为 null

  • 因为我们会把每层元素都放入到队列中， 如果队列为 null则代表二叉树已经遍历完了

• 创建一个 list集合保存元素 val

• 获取当前队列长度

  • 当前队列长度就代表着当前二叉树层级的节点个数

• 遍历获取二叉树当前层级的节点

  • 这里别忘记 size--；

• 判断当前节点的左右两个节点是否为 null， 不为 null则加入队列

代码展示

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        // 返回值
        List<List<Integer>> result = new ArrayList<>();
        // 如果如此那为 null，则直接返回
        if(root == null){
            return result;
        }
        // 设置队列
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer( root);
        // 循环遍历
        while(!queue.isEmpty()){
            // 创建当前层级的 list
            List<Integer> list = new ArrayList<>();
            // 获取当前层级的节点个数
            int size = queue.size();
            while(size > 0){
                // 获取当前节点
                TreeNode poll = queue.poll();
                if (poll.left != null){
                    queue.offer(poll.left);
                }
                // 将右节点放入队列中
                if (poll.right != null){
                    queue.offer(poll.right);
                }
                list.add(poll.val);
                size--;
            }
            // 将左节点放入队列中
            result.add(list);
        }
        return result;
    }
}
复制代码

结果分析

三、107.二叉树的层次遍历II

题目描述

解题思路

• 和上一题的唯一区别就是上一题是自顶上下遍历， 这道题是自下向上遍历
• 那么直接在上一题的基础上增加一个栈来存储每层节点元素
• 最后返回之前做一个弹栈操作就可以了， 具体可看下面代码展示

代码展示

提交结果

四、199.二叉树的右视图

题目描述

解题思路

• 这道题需要注意如下
  • 题中有说，是右视图，而不是获取最右节点
• 实际的解题方法还是在 102的基础上遍历每一层节点获取 list
• 在遍历 list获取每一层节点的最后一个节点
• 然后返回就可以了， 具体代码请看代码展示

代码展示

提交结果

五、637.二叉树的层平均值

题目描述

解题思路

• 一样的， 遍历每层节点的时候记录数值然后除一下就可以了
• 这里需要注意的是要用double接收

代码展示

提交结果

六、429.N叉树的层序遍历

题目描述

解题思路

• 这边需要注意的是二叉树变成了 n叉树， 其余不变
• 就是在我们添加每层节点的时候判断左右节点为 null变成了遍历添加就 ok

代码展示

提交结果

七、515.在每个树行中找最大值

题目描述

解题思路

• 这里需要注意一下节点的数值大小
• -231 <= Node.val <= 231 - 1

代码展示

提交结果

八、116.填充每个节点的下一个右侧节点指针

题目描述

解题思路

• 整体思路不变
• 遍历的时候将当前节点的 next指向下一个节点就可以了
• 但是需要注意的是最后一个节点指向 null， 所以判断条件为 size > 1 , 因为我们 size--， size == 1代表当前节点是该层节点的最后一个节点

代码展示

提交结果

九、117.填充每个节点的下一个右侧节点指针II

题目描述

解题思路

和上一题一样，可以直接代码复用

十、104. 二叉树的最大深度

题目描述

解题思路

• 整体思路不变，
• 每遍历一个层级 result++ 就可以了
• 111.二叉树的最小深度也是相同原理，就不讲解了

代码展示

提交结果