欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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);
		}
	}
}