【题32-2 分行从上到下打印二叉树】

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

【思路】
可以用一个队列来保存将要打印的节点，为了把二叉树的每一行单独打印到一行里，需要两个变量，一个变量表示在当前层还没有打印的节点数。一个表示下一层节点的数目。

【实现】

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

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

    }

}
*/
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null) return result;
        Queue<TreeNode> layer = new LinkedList<TreeNode>();
        ArrayList<Integer> layerList = new ArrayList<Integer>();
        layer.add(pRoot);
        int start = 0,end = 1;
        while(!layer.isEmpty()){
            TreeNode cur = layer.remove();
            layerList.add(cur.val);
            start++;
            if(cur.left != null){
                layer.add(cur.left);
            }
            if(cur.right!= null){
                layer.add(cur.right);
            }
            if(start == end){
                end = layer.size();
                start = 0;
                result.add(layerList);
                layerList = new ArrayList<Integer>();
            }
        }
        return result;
    }
}

参考：
1.《剑指offer》