我们都知道,广度优先遍历——对二叉树来说就是按层遍历,需要借助队列。代码也很简单,就几行。但是为什么要借助队列呢?粗鲁分析:
二叉树的按层遍历是这样:从左到右,从上到下访问每个节点。而二叉树本身给出的信息是其下一行的左右节点的,所以在访问某一行节点时,就要保存其含有的信息——下一行节点。
如果某行有n个节点,从1访问到n时,需要保存1的左右节点,2的左右节点,,,n的左右节点,遍历下行时的顺序是1的左右,2的左右,,,n的左右。
这不就是先进(先保存)先出(先访问)吗?
算法伪码:
1.root进队(每个元素都要进队,别直接访问)
2.开始循环:队首出队,其左右节点进队。(当然你也可以左右节点先进队,再队首出队,但是这样比较不舒服!——老子还没出去,儿子就进来了)
循环条件是队非空。
简单的代码:
public void visitTree(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
TreeNode cur;
while(!queue.isEmpty()) {
cur = queue.poll();//出队
System.out.println(cur.val);//访问这个出队的
if (cur.left != null) {
queue.offer(cur.left);
nlast = cur.left;
}
if (cur.right != null) {
queue.offer(cur.right);
nlast = cur.right;
}
}
}
=======================分割线==========