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

迭代：使用队列存储当前层节点 开始遍历。 遍历前 保存上一层节点数nodeCount, 遍历完后 上一层处理完毕，队列中剩余的即下一层所有节点。

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        // write code here
        if(root==null){
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int nodeCount = queue.size();
            ArrayList<Integer> sub = new ArrayList();
            list.add(sub);
            for(int i=0;i<nodeCount;i++){
                TreeNode treeNode = queue.poll();
                sub.add(treeNode.val);
                if(treeNode.left!=null){
                    queue.offer(treeNode.left);
                }
                if(treeNode.right!=null){
                    queue.offer(treeNode.right);
                }
            }
        }
        return list;
    }
}