【Leetcode】102.二叉树的层序遍历

题目

思路

我们要实现二叉树的层序遍历 要使用另一种数据结构——队列

层序遍历是打印完这一层 在打印下一层

我们从根节点开始遍历

直接将根节点压入队列中 然后打印出来 在出队列 判断根节点的左右节点是否为空 不为空继续压入队列

如果根节点左右都有节点的情况下 那么现在队列中保存着根节点的两个子节点 继续执行哦我们刚才的操作

直到队列为空 表明所有节点都已经被逐层遍历过了 层序遍历也就完成了

public void levelOrder(treeNode root){
        Queue<treeNode> queue = new LinkedList<>();
        if(root != null){
            queue.offer(root);
        }
        while (!queue.isEmpty()){
            treeNode top = queue.poll();
            System.out.println(top.val);
            if(top.left != null){
                queue.offer(top.left);
            }
            if(top.right != null){
                queue.offer(top.right);
            }
        }
    }

但是我发现题目要求我们返回值是一个嵌套的集合 类似于二维数组 每一层都单独放在一个集合中

这就表示我们不仅要遍历出来 还要将每层的元素甄别出来 放在同一个集合中 并最后放在一个总集合内

问题是 我们如何知道每一层有多少元素呢？
**我们可以定义一个size变量用来储存每一层拥有的元素 **
当我们将根节点插入队列中时 此时size = 1；
我们在将他的左节点和右节点插入时（假设都存在）size = 2；
这表示我们在下次操作时要操作两个头节点 将他们的左右节点压入栈中 这个操作的次数恰好就是操作前队列中的元素个数 因为压入队列的元素都相当于是一个头节点 我们都要对他的左右节点进行操作

这样我们的size就可以求出来了 size = 每一次小循环之后 队列中的元素个数

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        //队列
        List<List<Integer>> ret = new ArrayList<>();
        //返回的外部 存放内部集合
        if(root != null){
            queue.offer(root);
        }
        while(!queue.isEmpty()){
            List<Integer> back = new ArrayList<>();
            //内层集合 存放每一层的元素
            int size = queue.size();
            //求出每层元素的数目
            while(size != 0){
                TreeNode top = queue.poll();
                back.add(top.val);
                if(top.left != null){
                    queue.offer(top.left);
                }
                if(top.right != null){
                    queue.offer(top.right);
                }
                size--;
            }
            ret.add(back);
        }
        return ret;
    }
}