二叉树的层序遍历

一、题目

给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)

例如:

给定的二叉树是{3,9,20,null,null,15,7},

该二叉树层序遍历的结果是

\[3\],\[9,20\],\[15,7\]

二、解决方案

2.0 树节点构造

java 复制代码
 public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
 }

2.1 直接逐层遍历,定义nextNodeList和curLevelList,每遍历一层,就将当前层节点curList添加到结果res中。

java 复制代码
// 定义nextNodeList和curLevelList,每遍历一层,就将当前层节点curList添加到结果res中
    public List<List<Integer>> levelOrderOne(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null){
            return null;
        }

        // 存储当前层的节点
        List<TreeNode> curList = new ArrayList<>();
        curList.add(root);
        // 每遍历一层,就将当前层节点curList添加到结果res中
        while(!curList.isEmpty()){
            List<TreeNode> nextNodeList = new ArrayList<>();
            List<Integer> curLevelList = new ArrayList<>();
            for(TreeNode node : curList){
                curLevelList.add(node.val);
                if(node.left != null){
                    nextNodeList.add(node.left);
                }
                if(node.right != null){
                    nextNodeList.add(node.right);
                }
            }
            res.add(curLevelList);
            curList = nextNodeList;
        }
        return res;
    }

2.2 利用队列实现

因为队列是先进先出的数据结构,如果从左到右访问完一行节点,并在访问的时候把它们的子节点依次加入到队列,这时候它们的子节点也是按照顺序来进行的。且排在本行节点的后面,因此队列中出现的顺序正好也是从左到右,正好符合层次遍历的特点。

java 复制代码
    /**
     * 建立辅助队列,根节点首先进入队列。不管层次怎么访问,根节点一定是第一个,那它肯定排在队伍的最前面。
     * 每次进入一层,统计队列的元素个数,因为每当访问完一层,下一层作为这一层的子节点,一定都加入队列,而再下一层还没有加入,因此此时队列中的元素个数就是这一层的元素个数。
     * 每次遍历这一层这么多的节点数,将其依次从队列中弹出,然后加入这一行的一维数组中,如果它们有子节点,依次加入队列排队等待访问。
     * 访问完这一层的元素后,将这个一维数组加入二维数组中,再访问下一层。
     */
    public List<List<Integer>> levelOrderTwo(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null){
            return null;
        }

        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()){
            // 记录二叉树的某一行
            List<Integer> currentRow = new ArrayList();
            int currentQueueSize = queue.size();
            for (int i = 0; i < currentQueueSize; i++) {
                // 先存当前根节点
                TreeNode node = queue.poll();
                currentRow.add(node.val);
                // 若是左右孩子存在,则存入左右孩子作为下一个层次
                if(node.left!= null){
                    queue.add(node.left);
                }
                if(node.right!=null){
                    queue.add(node.right);
                }
            }
            res.add(currentRow);
        }

        return res;
    }
相关推荐
robin_suli13 分钟前
链表系列一>两两交换链表中的结点
数据结构·算法·链表
KyollBM35 分钟前
【CF】Day44——Codeforces Round 908 (Div. 2) C + Codeforces Round 1020 (Div. 3) DE
算法
努力学习的小廉39 分钟前
我爱学算法之—— 二分查找(上)
开发语言·c++·算法
小彭律师1 小时前
SinSR模型剪枝实验报告
人工智能·算法·剪枝
莹莹学编程—成长记3 小时前
string的模拟实现
服务器·c++·算法
ShiinaMashirol7 小时前
代码随想录打卡|Day27(合并区间、单调递增的数字、监控二叉树)
java·算法
wuqingshun3141599 小时前
蓝桥杯 5. 交换瓶子
数据结构·c++·算法·职场和发展·蓝桥杯
Demons_kirit9 小时前
Leetcode 2845 题解
算法·leetcode·职场和发展
adam_life10 小时前
http://noi.openjudge.cn/——2.5基本算法之搜索——200:Solitaire
算法·宽搜·布局唯一码
我想进大厂11 小时前
图论---朴素Prim(稠密图)
数据结构·c++·算法·图论