题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
先上代码,然后在分析。
import java.util.*;
/*
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> > result = new ArrayList<ArrayList<Integer> >();
if(pRoot == null){
return result;
}
ArrayList<Integer> temp = new ArrayList<Integer>();
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(pRoot);
int start = 0, end = 1;
while(!queue.isEmpty()){
TreeNode node = queue.poll();
temp.add(node.val);
start ++;
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
if(start == end){
end = queue.size();
start = 0;
result.add(temp);
temp = new ArrayList<Integer>();
}
}
return result;
}
}
分析如下:
利用队列先进先出的特点,先把根节点放入队列,然后弹出到list中,在依次访问左右节点,如果不为空,则依次放入队列。因为需要把每一层单独放在一个list中,所以需要使用end = queue.size();
记录每层节点的个数。每次弹出一个就执行start ++;
,当start == end
就代表这一层遍历结束,就把保存的list存入最终结果中,并且start = 0
,重新开始下一层的遍历。