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

题目描述

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

解题思路

1，用一个ArrayList<ArrayList<Integer>> 盛放所有行，每一行用一个ArrayList<Integer>
2，设置起始点，和重点，用来给当前行计数

代码

/**
 * 
 */
package offerTest;

import java.util.ArrayList;
import java.util.LinkedList;

/**
 * <p>
 * Title:Deep
 * </p>
 * <p>
 * Description:
 * </p>
 * 
 * @author 田茂林
 * @data 2017年8月21日 上午9:35:29
 */
public class Print {
    ArrayList<ArrayList<Integer>> ListPrint(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> listall = new ArrayList<ArrayList<Integer>>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        ArrayList<Integer> list = new ArrayList<Integer>();
        if (pRoot == null) {
            return listall;
        }
        int start = 0;
        int end = 1;
        int size = 0;
        queue.offer(pRoot);
        while (!queue.isEmpty()) {
            start++;
            TreeNode current = queue.poll();
            list.add(current.val);
            if (current.left != null) {
                queue.offer(current.left);
                size++;
            }
            if (current.right != null) {
                queue.offer(current.right);
                size++;
            }
            if (start == end) {
                start = 0;
                end = size;
                size = 0;
                listall.add(list);
                list = new ArrayList<Integer>();
            }
        }
        return listall;
    }
}