力扣102_二叉树的层次遍历

题目描述：
给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

算法思想：
看到层次遍历，立马想到了队列，想到了层次遍历框架

void bfs(TreeNode root) {
    LinkedList<TreeNode> queue = new LinkedList<>();
    queue.add(root);
    while (!queue.isEmpty()) {
        TreeNode p= queue.remove(); 
        if (p.left != null) {
            queue.add(p.left);
        }
        if (p.right != null) {
            queue.add(p.right);
        }
    }
}

可问题在于，如果按照这个框架，输出的就是[3,9,20,15,7]这样的数组，遍历刹不住车，怎么去区分哪层是哪层呢？
队列的访问一定是先进先出的，上一层的节点一定在下一层之前，如果我们在遍历这一层之前，做一个统计，知道这层有多少个节点，每层停顿一下，就能做出区分了。

代码实现：

public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new LinkedList<>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();   
        if(root!=null){
            queue.add(root);
        }     
        
        while(!queue.isEmpty()){
            int size = queue.size();
            LinkedList<Integer> level = new LinkedList<>();
            //循环size次，分层
            for(int i = 0; i < size; i++){
                TreeNode p = queue.remove();
                level.add(p.val);
                if(p.left!=null){
                    queue.add(p.left);
                }
                if(p.right!=null){
                    queue.add(p.right);
                }
                       
            }
            res.add(level);
        }
        return res;
    }