剑指offer-把二叉树打印成多行-java

题目描述：

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

思路解析：

使用队列，层序遍历数组，同样采用一个元素出队后，就把他的左右孩子加进来，不同的是需要计数每一层的元素个数

代码：

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    /*跟层序遍历一个思想，队列中去掉一个就要把他的左右子节点加进来
    不同的是层序遍历一层一个list，所以需要计数
    */
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
        if(pRoot ==null)
            return listAll;
        Queue<TreeNode> q = new LinkedList<TreeNode>();//Queue是接口，LinkedList是其实现类
        q.add(pRoot);
        while(!q.isEmpty()){
            int last = q.size(),count=0;//last:每一层的数字；count：每一层存储到第几个了。
            ArrayList<Integer> list = new ArrayList<Integer>();
            while(count<last){
                TreeNode node = q.poll();
                list.add(node.val);
                count++;
                if(node.left!=null)
                    q.add(node.left);
                if(node.right!=null)
                    q.add(node.right);
            }
            listAll.add(list);
        }
        return listAll;
    }
    
}