题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
1、思路分析
因为每层的结点均是从左往右打印,所以只需要用一个队列来进行存储即可,需要注意的是,题目要求每一层输出一行,为了区别不同的行数,需要记录当前层和下一层的结点数。采取添加-移除-添加的方式。
2、代码

public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        ArrayList<Integer> tmp = new ArrayList<>();
        LinkedList<TreeNode> tn = new LinkedList<>();
        if(pRoot == null) {
            return res;
        }
        tn.add(pRoot);
        int now = 1, next = 0;
        while(!tn.isEmpty()) {
            TreeNode t = tn.remove();
            now--;
            tmp.add(t.val);
            if(t.left != null) {
                tn.add(t.left);
                next++;
            }
            if(t.right != null) {
                tn.add(t.right);
                next++;
            }
            if(now == 0) {
                res.add(new ArrayList<Integer>(tmp));
                tmp.clear();
                now = next;
                next = 0;
            }
        }
        return res;
    }
}