把二叉树打印成多行

题目描述

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

解题思路

用队列存储每个结点，用两个指针，一个指针表示移动的指针，另一个指针表示这一行的最后的结点，每次从队列拿出的指针如果是这一行的最后一个结点的话那么就把数据存到ArrayList<ArrayList<Integer>>，所以这么一来的话就可以达到层次遍历。

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<>();
        Queue<TreeNode> queue = new LinkedList<>();
        ArrayList<Integer> list = new ArrayList<>();
        if(pRoot == null) return listAll;
        queue.offer(pRoot);
        TreeNode last = pRoot;
        TreeNode nlast = null;
        while(!queue.isEmpty()){
            pRoot = queue.poll();
            list.add(pRoot.val);
            if(pRoot.left != null){
                queue.add(pRoot.left);
                nlast = pRoot.left;
            }
            if(pRoot.right != null){
                queue.add(pRoot.right);
                nlast = pRoot.right;
            }
            if(last == pRoot){
                last = nlast;
                listAll.add(list);
                list = new ArrayList<>();
            }
        }
        return listAll;
    }
}

以上就是这道题的解法。