将数组存入二叉树

最新推荐文章于 2024-10-17 20:11:22 发布

原创最新推荐文章于 2024-10-17 20:11:22 发布 · 1.5k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#java

这篇博客讲述了如何将数组中的元素构建为一棵二叉树，并通过递归实现前、中、后序遍历。首先，将数组的第一个元素作为根节点，然后循环遍历数组，根据元素与当前节点值的大小关系决定将其添加到左子树还是右子树。在遍历过程中，采用中序遍历可以得到有序序列。

将数组存入二叉树

手工创建二叉树并实现前、中、后序排列很简单，当尝试将数组存入二叉树时，感觉就没那么容易了，当时分析得很简单就是在数组中任意取一个数作为根结点，然后将剩余的数与该结点进行比较，比根结点小的放到左结点上，大的放到右结点上，我本想用一次循环将所有的数字就各归其位，在一次比较中将所有的数分成了两类，一类比根结点大一类比根结点小，虚拟机将它们赋值时就会无所是从。总想将所有工作一次完成，于是就磕死在这个地方了。

真正要实现将数组存入二叉树我的实现方法是：先将数组中的第一个元素作为根结点的数据

int n = array[0];

root = new Node(n);

然后循环调用向树中加入数据的方法，并向其中传入结点和数组中数据：

for(int i=1;i<array.length;i++){

addTotree(root,array[i]);

}

在将数组中数据加入到树的相应位置时应先创建一个结点对象

// 把传入的数创建为一个二叉树的节点

Node newnode=new Node(data);

然后进行判断传入的数组中的数据与传入结点中的数据的大小，若传入的数据比根结点小且当左子树不为空则递归为空则加到左结点

if((Integer)node.getObj()>data){

// 如果左结点不为空

if(node.getLeft()!=null){

// 将 node 左结点赋给 node

node = node.getLeft();

// 递归

addTotree(node,data);

}else{

node.setLeft(newnode);

newnode.setParent(node);

}

加到右子树的方法同样。

这样也就将数组全部加到二叉树中了，打印时同样用到了递归

/**

* 遍历打印二叉树中的元素

* @param root

public void printTree(Node root) {

// 中序遍历

if (root != null) {

printTree(root.getLeft());

Object obj = root.getObj();

System.out.println(obj);

printTree(root.getRight());

}

按照中序遍历即可得到按顺序排列的一组数据。

package javastudy.caidan.二叉树0810;

/**
 * 二叉树类
 * 
 * @author 蔡丹
 * 
 */
public class Tree {
	private Node root;
	private Node node;
	/**
	 * 程序入口
	 * @param args
	 */
	public static void main(String args[]) {
		// 创建一个二叉树对象
		Tree tr = new Tree();
		int length=10;
		int [] array=tr.createSrcArray(length);
		for(int i=0;i<array.length;i++){
			System.out.print(array[i]+"\t");
		}
		System.out.println("\n");
		// 创建一个根结点
		Node root = tr.arrayTotree(array);
		// 遍历打印二叉树
		tr.printTree(root);
	}
	public int[] createSrcArray(int len){
		// 初始化要排序数组中的值
		int[] array = new int[len];
		for(int i=0;i<array.length;i++){
			// 创建一个随即对象
			java.util.Random ran = new java.util.Random();
			// 调用随即对象，每次循环时生成一个0~100间随机数
			int value = ran.nextInt(100);
			// 给数组中指定位置填上随机数
			array[i] = value;
			
		}
		return array;
	}
	/**
	 * 将数组中的数传入树中	
	 * @param array
	 * @return
	 */
	public Node arrayTotree(int[] array){
		// 如果为空抛出异常
		if(array.length==0){
			throw new RuntimeException("<<<<<<<<<<");
		}
		else {
			int n = array[0];
			 root = new Node(n);
			 for(int i=1;i<array.length;i++){
				 addTotree(root,array[i]);
			 }
		}		
		return root;
	}
	
	/**
	 * 将数组中的数存到树中相应位置
	 * @param node
	 * @param array
	 */
	public void addTotree(Node node,int data){
		
		//把传入的数创建为一个二叉树的节点
		Node newnode=new Node(data);		
		if((Integer)node.getObj()>data){
			// 如果左结点不为空
			if(node.getLeft()!=null){
				// 将node左结点赋给node
				node = node.getLeft();
				// 递归
				addTotree(node,data);
				
			}else{
				
				node.setLeft(newnode);
				newnode.setParent(node);
			}
			
		}
		else{
			// 如果结点为空
			if(node.getRight()!=null){
				//将该结点设为右结点
				node = node.getRight();
				// 递归
				addTotree(node,data);
			}else{
				node.setRight(newnode);
				newnode.setParent(node);
			}
		}		
	}
	/**
	 * 遍历打印二叉树中的元素
	 * @param root
	 */
	public void printTree(Node root) {
		// 中序遍历
		if (root != null) {
			printTree(root.getLeft());
			Object obj = root.getObj();
			System.out.println(obj);
			printTree(root.getRight());
		}
	}
}