题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
代码
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/*
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>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(pRoot == null) return res;
queue.add(pRoot);
while(!queue.isEmpty()){
int size = queue.size();//用size来控制本层的节点数量
ArrayList<Integer> list = new ArrayList<>();
while(size != 0){
TreeNode node = queue.poll();
list.add(node.val);
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
size--;//容易忘记size--
}
res.add(list);//跳出上一个while循环一层的都已经加入到list了,再把list加入res
}
return res;
}
}