java树的广度优先遍历思路
程序员文章站
2022-05-23 11:28:28
...
我们使用广度优先遍历的办法, 来 遍历一棵树,我们就需要记录下每一层的节点,从每一层的节点中去获取下一层节点,所有我们需要有一个记录每层节点的容器来完成这个工作,简单点的办法就是使用队列,注意队列数据的顺序;
创建树的方法:
public static TreeNode createTree(){
TreeNode firstLeft=new TreeNode(3);
TreeNode firstRighe=new TreeNode(7);
TreeNode root=new TreeNode(5,firstLeft,firstRighe);
TreeNode leftSecondL=new TreeNode(2);
TreeNode rightSecondL=new TreeNode(4);
TreeNode leftSecondR=new TreeNode(6);
TreeNode rightSecondR=new TreeNode(8);
firstLeft.setLeftNode(leftSecondL);
firstLeft.setRightNode(rightSecondL);
firstRighe.setLeftNode(leftSecondR);
firstRighe.setRightNode(rightSecondR);
return root;
}
package com.czy.tree;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
/*
* 深搜和广搜
*/
public class TreeSerch {
public static List wideSerch(TreeNode root){
List<Integer> list=new ArrayList<Integer>(); //链表用来存数据
Queue<TreeNode> queue=new LinkedList<TreeNode>();//队列
if(root==null){
return list;
}
queue.offer(root);//先把根节点入队,使队列不为空,进入循环
while(!queue.isEmpty()){
TreeNode node=queue.poll();
if(node.getLeftNode()!=null)
queue.offer(node.getLeftNode());
if(node.getRightNode()!=null)
queue.offer(node.getRightNode());
list.add(node.getData());
}
return list;
}
public static void main(String[] args) {
TreeNode root=TestBstTree.createTree();//调用了自己写的静态方法创建一棵树
List<Integer> list=wideSerch(root);
for(Integer one:list){
System.out.println(one);
}
}
}