面试准备之--二叉树打印

题目：

有一棵二叉树，请设计一个算法，按照层次打印这棵二叉树。

给定二叉树的根结点root，请返回打印结果，结果按照每一层一个数组进行储存，所有数组的顺序按照层数从上往下，且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。

import java.util.*;
 
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class TreePrinter {
  public static int[][] printTree(TreeNode root) {
        if(root==null)
            return null;
            //用一个队列来装要打印的数据
        LinkedList<TreeNode> link=new LinkedList<TreeNode>();
        //结果存在数组中
        ArrayList<ArrayList<Integer>> layer=new ArrayList();
        ArrayList<Integer> arr=new ArrayList<Integer>();
        //设置两个指针一个last一个nlast
        //首先让last指向根节点，根节点进入队列
        TreeNode last=root;
        //nlast指向下一行最右节点
        TreeNode nlast=null;
        //根节点进入队列
        link.add(root);
         
        while(!link.isEmpty()){
            TreeNode tem=link.removeFirst();
            //nlast=tem;
            arr.add(tem.val);
            if(tem.left!=null){
                link.add(tem.left);
                nlast=tem.left;
            }
            if(tem.right!=null){
                link.add(tem.right);
                nlast=tem.right;
            }
            if(tem==last){
                layer.add(arr);
                arr=new ArrayList<Integer>();
                last=nlast;
            }
        }
         
        int num[][]=new int[layer.size()][];
        for(int i=0;i<layer.size();i++){
            num[i]=new int[layer.get(i).size()];
            for(int j=0;j<layer.get(i).size();j++){
                num[i][j]=layer.get(i).get(j);
            }
        }
         
        return num;
      }
}