需要用到队列Queue来存储节点对象,先进先出
先往队列中插入左节点,再插右节点,这样出队就是先左节点后右节点了。
广度优先遍历树,需要用到队列(Queue)来存储节点对象,队列的特点就是先进先出。例如,上面这颗树的访问如下:
首先将A节点插入队列中,队列中有元素(A);
将A节点弹出,同时将A节点的左、右节点依次插入队列,B在队首,C在队尾,(B,C),此时得到A节点;
继续弹出队首元素,即弹出B,并将B的左、右节点插入队列,C在队首,E在队尾(C,D,E),此时得到B节点;
继续弹出,即弹出C,并将C节点的左、中、右节点依次插入队列,(D,E,F,G,H),此时得到C节点;
将D弹出,此时D没有子节点,队列中元素为(E,F,G,H),得到D节点;
。。。以此类推。。
代码:这里以二叉树为例,遍历所有节点的值
package sjm;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
//广度搜索算法
class TreeNode{
int val=0;
TreeNode left=null;
TreeNode right=null;
public TreeNode(int val){
this.val=val;
}
}
public class Two {
/**
* @param args
*/
public static ArrayList<Integer> PrintFromTopToBottom(TreeNode root){
ArrayList<Integer> lists=new ArrayList<Integer>();
if(root==null){
return lists;
}
Queue<TreeNode> queue=new LinkedList<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode tree =queue.poll();
if(tree.left!=null){
queue.offer(tree.left);
}
if(tree.right!=null){
queue.offer(tree.right);
}
lists.add(tree.val);
}
return lists;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
TreeNode tr1=new TreeNode(1);
TreeNode tr2=new TreeNode(3);
TreeNode tr3=new TreeNode(2);
TreeNode tr4=new TreeNode(6);
TreeNode tr5=new TreeNode(7);
TreeNode tr6=new TreeNode(8);
TreeNode tr7=new TreeNode(0);
tr1.left=tr2;
tr1.right=tr3;
tr2.left=tr4;
tr2.right=tr5;
tr3.left=tr6;
tr3.right=tr7;
List<Integer> test=new ArrayList<Integer>();
test=PrintFromTopToBottom(tr1);
for(int i:test){
System.out.print(i+" ");
}
}
}
输出结果:
1 3 2 6 7 8 0