把二叉树打印成多行

题目描述

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

//思路：也是每一层放到队列里然后出队。

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>> list = new ArrayList<>();
        if(pRoot == null){
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(pRoot);
        int res = 0;
        while(!queue.isEmpty()){
            ArrayList<Integer> list2 = new ArrayList<>();
            res++;
            int n = queue.size();
            for(int i = 0 ; i < n; i++){
                pRoot = queue.poll();
                list2.add(pRoot.val);
                if(pRoot.left != null){
                    queue.add(pRoot.left);
                }
                if(pRoot.right != null){
                    queue.add(pRoot.right);
                }
            }
            list.add(list2);
        }
        return list;
            
            
    }
    
}