8力扣题做题笔记及复盘--102.二叉树层序遍历(二叉树，队列)

102. 二叉树的层序遍历 - 力扣（LeetCode）

1、题外话

层序遍历一个二叉树。就是从左到右一层一层的去遍历二叉树。这种遍历的方式和我们之前讲过的都不太一样。

需要借用一个辅助数据结构即队列来实现，队列先进先出，符合一层一层遍历的逻辑。

2、思路

分两步走吧，先能够获取一个层序遍历后的集合，再考虑对集合进行分层，实现题目要求的效果

①二叉树的层序遍历：

1、判断队列是否为空

2、从头部弹出一个元素

3、获取头部元素的左右子节点并添加进队列。

②分段：

每次添加之后形成一个新的队列，对队列进行输出添加到集合中，小集合再添加到大集合中

3、代码

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
    
        List<List<Integer>> list = new ArrayList<>();

        Queue<TreeNode> queue = new LinkedList<>();

        if(root == null){
            return list;
        }
        queue.offer(root);
//每一轮会产生一个新队列，对队列进行判断是否为空。
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            System.out.println(node.val);
            if(node.left !=null){
                queue.offer(node.left);
            }
            if(node.right !=null){
                queue.offer(node.right);
            }
        }

        return null;
    }
}

 

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {

         List<List<Integer>> res = new ArrayList<>();

         Queue<TreeNode> queue = new LinkedList<>();
//注意边界，当root为空时，返回一个空的集合。
         if (root == null) return res;

        queue.offer(root);
//先考虑如何用一个集合存储层序遍历过来的数据，再考虑集合分段
//集合分段：通过每个队列的大小遍历元素添加到小集合中，每个元素的左右子节点添加到下次队列中
        while(!queue.isEmpty()){
          
           int size = queue.size();
           List<Integer> list = new ArrayList<>();
//对旧队列进行循环获取
            for(int i=0;i<size;i++){
                TreeNode node = queue.poll();
//添加进新队列中
                if(node.left!=null){
                     queue.offer(node.left);
                }
                if(node.right !=null){
                    queue.offer(node.right);
                }
                list.add(node.val);
            }
           res.add(list);
           
        }
        return res;
    }
}

4、总结

层序遍历，用队列去存储，然后对队列进行操作，队头弹出的元素放在集合中。

如何分段，充分利用size大小进行for循环切分。队列执行添加后就是新队列的大小。