【树11】把二叉树打印成多行

题目描述

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

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 {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        //按行保存
        ArrayList<ArrayList<Integer> > nodesList=new ArrayList<ArrayList<Integer> >();
        //保存每行的结点
        ArrayList<Integer>  list=new ArrayList<Integer>();
        if(pRoot==null)
            return nodesList;
        //下一层的结点数
        int nextLevelNodes=0;
        //下一行要打印的结点数
        int toBePrint=1;
        Queue<TreeNode> queue=new LinkedList<TreeNode>();
        queue.add(pRoot);
        while(!queue.isEmpty()){
           
            TreeNode tempNode=queue.poll();
            list.add(tempNode.val);
            if(tempNode.left!=null){
                queue.add(tempNode.left);
                ++nextLevelNodes;
            }
            if(tempNode.right!=null){
                queue.add(tempNode.right);
                ++nextLevelNodes;
            }
            --toBePrint;
            //当前层结点打印全部后
            if(toBePrint==0){
                nodesList.add(list);
                toBePrint=nextLevelNodes;
                //清0，重新统计下一行的结点数
                nextLevelNodes=0;
                //清空集合
                list=new ArrayList<Integer>();
            }
        }
        return nodesList;
        
        
        
        
        
        
        
    }
    
}