二叉树的层序遍历

感谢carl，感谢代码随想录

对于二叉树来说，有深度遍历优先和广度遍历优先，深度遍历可以使用栈的方式进行，而广度优先遍历更符合队列。

1.题目引出

力扣题目链接

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

层序遍历二叉树，就是从左到右一层层的去遍历二叉树。

我们可以使用一个辅助队列来完成。队列先进先出，符合一层一层遍历的逻辑，而是用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。

public void doQueue(List<List<Integer>> res, TreeNode root) {
    if (root == null) return; //如果头节点为空，返回
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root); //先将头节点放入队列

    //每次循环就会处理一层，
    while (!queue.isEmpty()) {
        List<Integer> list = new ArrayList<>();
        int len = queue.size(); //当前层数有多少节点

        //依次处理该层的所有节点
        while(len-- > 0) {
            TreeNode tmp = queue.poll(); //弹出队首
            list.add(tmp.val);
            if (tmp.left != null) queue.offer(tmp.left); //如果此节点有左孩子，加入队尾
            if (tmp.right != null) queue.offer(tmp.right); //如果此节点有右孩子，加入队尾
        }

        res.add(list);
    }
}

2.模板建立

从上面我们就可以推导出对树进行层序遍历的模板

Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root); //先将头节点放入队列

//每次循环就会处理一层，
while (!queue.isEmpty()) {
    int len = queue.size(); //当前层数有多少节点

    //逻辑处理
    
    for (int i = len; i > 0; i--) {
        //逻辑处理
        
        TreeNode tmp = queue.poll(); //弹出队首
        
        //逻辑处理

        if (tmp.left != null) queue.offer(tmp.left); //如果此节点有左孩子，加入队尾
        if (tmp.right != null) queue.offer(tmp.right); //如果此节点有右孩子，加入队尾
    }
}