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

60. 把二叉树打印成多行

题目描述

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

层序遍历即可

import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null){
            return list;
        }
        // 层序遍历
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(pRoot);
        TreeNode top;
        while(!queue.isEmpty()){
            int count = queue.size();
            ArrayList<Integer> innerList = new ArrayList<Integer>();
            for(int i = 0; i < count; i++){
                top = queue.poll();
                innerList.add(top.val);
                if(top.left != null)
                    queue.offer(top.left);
                if(top.right != null)
                    queue.offer(top.right);
            }
            list.add(innerList);
        }
        return list;
    }
    
}

 注意：

虽然使用下面的语法也能转化成int[]数组，但是效率会比我们手动使用循环来转换低很多

int[] arr = list.stream().mapToInt(Integer::valueOf).toArray();    // 虽然也能转换成int数组，但是效率低了很多