java实现二叉树的层次遍历

昨天面试的时候遇到这个问题，由于太久没有接触算法和数据结构了，导致遗忘的比较彻底，当时记得使用队列的特性可以实现层次遍历的问题，但是自己写的时候却没有很好的利用队列的特性，当时的第一想法是递归，最后写着写着就因为找不到好的出口条件，以及各种分支判断，加上如果有兄弟子树，还要写如何处理这种情况等等，最后反正是写不下去了。

所以今天着重记录一下，如何正确的使用队列的特性去实现二叉树的层次遍历。当然这次的状况也很好的警示：一般情况下，不要使用递归，除非这个递归的出口以及内部条件及其容易做出判断。所以这种东西还是需要经验的积累。

接下来看如何正确实现吧。参考blog

public class LevelOrder {
public void levelIterator(BiTree root){
if (root == null){
return;
}
LinkedList<BiTree> queue = new LinkedList<BiTree>();
BiTree current = null;
queue.offer(root); // 根节点入队
while (!queue.isEmpty()){ // 只要队列中有元素，就可以一直执行，非常巧妙的利用了队列的特性
current = queue.poll(); // 出队队头元素
System.out.print("-->" + current.data);
// 左子树不为空，入队
if (current.leftChild != null)
queue.offer(current.leftChild);
// 右子树不为空，入队
if (current.rightChild != null)
queue.offer(current.rightChild);

}
}

public static void main(String[] args) {
BiTree b1 = new BiTree();
b1.setData(1);
BiTree b2 = new BiTree();
b2.setData(2);
BiTree b3 = new BiTree();
b3.setData(3);
BiTree b4 = new BiTree();
b4.setData(4);
BiTree b5 = new BiTree();
b5.setData(5);
BiTree b6 = new BiTree();
b6.setData(6);
BiTree b7 = new BiTree();
b7.setData(7);

b1.setLeftChild(b2);
b1.setRightChild(b3);
b2.setLeftChild(b4);
b2.setRightChild(b5);
b3.setLeftChild(b6);
b3.setRightChild(b7);

LevelOrder order = new LevelOrder();
order.levelIterator(b1);
}
}

最后放两张图，大致理解一下吧，我就不说了。