题解 | #求二叉树的层序遍历#

import java.util.*;



public class Solution {
   
    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        // write code here
        ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
        if(root == null){
            return ret;
        }
        
        //用队列辅助，存储一层元素，根据这一层元素得到下一层元素
        Queue<TreeNode> q = new ArrayDeque<>();
        q.add(root);
        //当队列不为空时
        while(!q.isEmpty()){
            ArrayList<Integer> tmp = new ArrayList<>();
            //得到q的大小（一层的个数），遍历完这一层加入到列表
            // 同时将下一层的节点加入队列
            int n =q.size();
            for(int i=0;i<n;i++){
                TreeNode t = q.poll();
                tmp.add(t.val);
                //同时将这一个节点的子节点加入队列
                //以下新增加的节点与n无关属于下一层
                if(t.left!=null){
                    q.add(t.left);
                }
                if(t.right!=null){
                    q.add(t.right);
                }
            }
            //将这一层加入ret
            ret.add(tmp);

        }


        return  ret;
    }
}