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

1、题目描述

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

2、思路

设置3个变量，分别是cur，last，nlast，cur表示的是当前节点号，last表示的是当前行最后一个节点号，nlast表示的是下一行的最后一个节点号。当cur等于last时，换行，将last重新置为nlast，将nlast置为0，根据加入队列的节点数重新计算下一行的节点数，即nlast。重复上述步骤，知道队列为空为止。

3、代码实现

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>> listAll = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null){
            return listAll;
        }
        int last = 1;//指向当前行的最后一个节点号
        int nlast = 0;//指向下一行的最后一个节点号
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.offer(pRoot);//先将根节点压入队列
        int cur = 0;//指向当前节点号

        //统计下一行的节点数量，确定下一行的最后一个节点号
        if(pRoot.left != null){
                
                nlast++;
                
        }
        if(pRoot.right != null){
               
                nlast++;
        }                                      
         ArrayList<Integer> list = new ArrayList<Integer>();    
        
        while(!q.isEmpty()){
           
            TreeNode node = q.poll();
            list.add(node.val);
             cur++;
             if(cur == last){//因为cur是从0开始
                listAll.add(list);
                list = new ArrayList<Integer>();
                last = nlast;
                cur = 0;
                nlast = 0; //重新统计下一行的节点数量，确定下一行最后一个节点的行号
            }
            if(node.left != null){
                q.offer(node.left);
               
                nlast++;
                
            }
            if(node.right != null){
                q.offer(node.right);
                
                nlast++;
            }
 
            
        }
   
     return listAll;
    
  }
    
}