把二叉树打印成多行

题目

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

 

分析

　　用队列从根开始存储，首先判断队列是否为空，若不为空，取出第一个元素，将该元素的左孩子和右孩子（在有的情况下）继续存到队列中，知道队列为空。

　　注，因为返回的是ArrayList<ArrayList<Integer>>类型，所以我们应该注意在哪里去创建ArrayList<Integer>。

 

代码

    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        if(pRoot==null)
            return list;
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.add(pRoot);
        while(!q.isEmpty()){
            int low = 0, high = q.size();
            ArrayList<Integer> li = new ArrayList<Integer>();
            while(low<high){
                TreeNode node = q.poll();
                li.add(node.val);
                if(node.left!=null)
                    q.add(node.left);
                if(node.right!=null)
                    q.add(node.right);
                low++;
            }
            list.add(li);
        }
        return list;
    }