题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路
①: 该题为二叉树的层序遍历;需要借助一个队列,队列具有先进先出的特性;
②: 在java中LinkedList实现了Deque接口,Deque<E> extends Queue<E>;
③: 队列的一些方法,进队(offer),队头元素(peek),出队并返回(poll);
④: 我们可以先将root节点入队,当队列不空时,先出队(打印),判断该元素的左节点是否为null,如果左不为null,左入队,判断右,是否为null,如果右不为null,右入队;
代码实现
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<Integer>();
Queue<TreeNode> queue = new LinkedList<TreeNode>();
if(root == null){
return list;
}else{
queue.offer(root);
}
while(!queue.isEmpty()){
//出队
TreeNode node = queue.poll();
//打印
list.add(node.val);
//左不空左入队
if(node.left!=null){
queue.offer(node.left);
}
//右不空右入队
if(node.right!=null){
queue.offer(node.right);
}
}
return list;
}