从上往下打印出二叉树的每个节点,同层节点从左至右打印。
规律:每一次打印一个节点的时候,如果该节点有子节点,则把该节点的子节点放到一个队列的尾部。接下来到对队列的头部取出最早进入队列的节点放到ArrayList 中,重复前面的操作,直至队列中所有的节点都存到ArrayList中。
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
//queue用来保存当前遍历到了哪个节点,一次性把一个节点的左右子都入队
Queue<TreeNode> queue = new LinkedList<TreeNode>();
//list用来保存输出的节点
ArrayList<Integer> list = new ArrayList<Integer>();
if(root==null){//注意:空树返回一个默认构造的空LinkedList,而不是一个空指针null
return list;
}
TreeNode current = root;
queue.offer(current);
//只要队列中还有节点就说明还没遍历完,继续。
//每次从队列出队,然后将这个节点左右子入队列(FIFO,故能完成广度/层级遍历),再将这个节点记录在list中即可。
while(!queue.isEmpty()){
current = queue.poll();
list.add(current.val);
if(current.left!=null){//有左子则入队
queue.offer(current.left);
}
if(current.right!=null){
queue.offer(current.right);
}
}
return list;
}
}