从上往下打印二叉树
问题:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
分析:
层序遍历直接想到的应该是基于队列实现。从上到下打印二叉树的规律:每一次打印一个结点的时候,如果该结点有子结点,则把该结点的子结点放到一个队列的末尾。接下来到队列头部取出最早进入队列的结点,重复前面的打印操作,直到队列中所有的结点都被打印出来为止。以下面例子说明:
按层打印二叉树过程:
注意不管是广度优先遍历一个有向图还是一颗树,都会用到队列。第一步我们把起始结点(对树而言是根结点)放入到队列中。接下来每一次从队列的头部取出一个结点,遍历这个结点之后把从它能到达的结点(对数而言是子结点)都依次放入队列。我们重复这个遍历过程,直到队列中的结点全部被遍历为止。
编写代码如下:
importjava.util.ArrayList;
importjava.util.LinkedList;
publicclass Solution {
public ArrayListPrintFromTopToBottom(TreeNode root) {
if(root==null)//先考虑边界情况
return newArrayList();
//初始化list和queue
ArrayList list=newArrayList();
LinkedList queue=newLinkedList();
//现将根结点入队列
queue.offer(root);
while(!queue.isEmpty()){
TreeNode treeNode=queue.poll();
list.add(treeNode.val);
if(treeNode.left!=null)
queue.offer(treeNode.left);
if(treeNode.right!=null)
queue.offer(treeNode.right);
}
return list;
}
}