题目
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路
上一题精简版,ArrayList<ArrayList >已经体现了按层次打印
代码
import java.util.*;
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer> > result = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(pRoot == null) return result;
queue.offer(pRoot);
while(!queue.isEmpty()){
int size = queue.size();
ArrayList<Integer> list= new ArrayList<>();
for(int i = 0 ; i< size; i++){
TreeNode node = queue.poll();
list.add(node.val);
if(node.left != null) queue.offer(node.left);
if(node.right != null) queue.offer(node.right);
}
result.add(list);
}
return result;
}
}