1. 程式人生 > >二叉排序樹的建立和遍歷

二叉排序樹的建立和遍歷

   輸入一系列整數,建立二叉排序樹,並進行前序、中序、後序遍歷。

#include <iostream>
using namespace std;
#include <stdio.h>
#include <stdlib.h>

typedef struct BiTNode
{
	int value;
	struct BiTNode *lchild,*rchild;
}*BiTree;

bool LT(int a,int b)  //LessThan小於
{
	if(a<b)
		return true;
	else
		return false;
}
/*
在根指標root所指向的二叉排序樹中遞迴地查詢其關鍵字等於data的資料元素,若查詢成功,則指標p指向該資料元素結點,並返回true,
否則指標p指向查詢路徑上訪問的最後一個結點並返回false指標,指標f指向root的雙親,其初始呼叫值NULL
*/
bool SearchBST(BiTree root,int data,BiTree f,BiTree &p)
{
	if(!root)
	{
		p=f;
		return false;
	}
	else if(data==root->value)
	{
		p=root;
		return true;
	}
	else if(data<root->value)
		return SearchBST(root->lchild,data,root,p);
	else if(data>root->value)
		return SearchBST(root->rchild,data,root,p);
}

//當二叉排序樹root中不存在關鍵字等於data的資料元素時,插入data
inline void InsertBST(BiTree &root,int data)     //root為傳引用指標
{  
	BiTree p,s;
	if(!SearchBST(root,data,NULL,p))    //查詢不成功
	{
		s=(struct BiTNode *)malloc(sizeof(BiTNode));
		s->value=data;
		s->lchild=s->rchild=NULL;
		if(p==NULL)    //二叉排序樹為空的時候,被插入結點*s為新的根結點
			root=s;
		else if(LT(data,p->value))           //被插結點*s為左孩子
			p->lchild=s;
		else           //被插結點*s為右孩子
			p->rchild=s;
	}
	return ;
}
void PreOrderTraverse(BiTree root)    //先序遍歷
{
	if(root)
	{
		printf("%d ",root->value);
		PreOrderTraverse(root->lchild);
		PreOrderTraverse(root->rchild);
	}
}
void InOrderTraverse(BiTree root)    //中序遍歷
{
	if(root)
	{
		InOrderTraverse(root->lchild);
		printf("%d ",root->value);
		InOrderTraverse(root->rchild);
	}
}
void PostOrderTraverse(BiTree root)    //後序遍歷
{
	if(root)
	{
		PostOrderTraverse(root->lchild);
		PostOrderTraverse(root->rchild);
		printf("%d ",root->value);
	}
}
void DeleteBST(BiTree root)
{
	if(root)
	{
		DeleteBST(root->lchild);    //釋放左子樹
		DeleteBST(root->rchild);    //釋放右子樹
		free(root);        //釋放根結點
	}
}
int main(void)
{
	int i,a[101],n;
	BiTree root;
	while(scanf("%d",&n)!=EOF)
	{
		root=NULL;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			InsertBST(root,a[i]);
		}
		PreOrderTraverse(root);
		printf("\n");
		InOrderTraverse(root);
		printf("\n");
		PostOrderTraverse(root);
		printf("\n");
		DeleteBST(root);
	}
	return 0;
}