C实现二叉树简单算法

C实现二叉树简单算法

//二叉树的基本运算算法
#include <stdio.h>
#include <malloc.h>
#define MaxSize 100
typedef char ElemType;
typedef struct node 
{	
	ElemType data;			//数据元素
	struct node *lchild;	//指向左孩子结点
	struct node *rchild;	//指向右孩子结点
} BTNode;
void CreateBTree(BTNode * &b,char *str)	//创建二叉树
{
	BTNode *St[MaxSize],*p=NULL;
	int top=-1,k,j=0;  
	char ch;
	b=NULL;				//建立的二叉树初始时为空
	ch=str[j];
	while (ch!='\0')  	//str未扫描完时循环
	{
   	   	switch(ch) 
		{
		case '(':top++;St[top]=p;k=1; break;		//为左孩子结点
		case ')':top--;break;
		case ',':k=2; break;                      		//为孩子结点右结点
		default:p=(BTNode *)malloc(sizeof(BTNode));
				p->data=ch;p->lchild=p->rchild=NULL;
				if (b==NULL)                    	 	//*p为二叉树的根结点
					b=p;
				else  								//已建立二叉树根结点
				{	
					switch(k) 
					{
					case 1:St[top]->lchild=p;break;
					case 2:St[top]->rchild=p;break;
					}
				}
		}
		j++;
		ch=str[j];
	}
}
void DestroyBTree(BTNode *&b)	//销毁二叉树
{	if (b!=NULL)
	{	DestroyBTree(b->lchild);
		DestroyBTree(b->rchild);
		free(b);
	}
}

/*

BTNode *FindNode(BTNode *b,ElemType x) //查找值为x的结点
{
	BTNode *p;
	if (b==NULL)
		return NULL;
	else if (b->data==x)
		return b;
	else  
	{
		p=FindNode(b->lchild,x);
		if (p!=NULL) 
			return p;
		else 
			return FindNode(b->rchild,x);
	}
}
BTNode *LchildNode(BTNode *p)
{
    return p->lchild;
}
BTNode *RchildNode(BTNode *p)
{
    return p->rchild;
}
int BTHeight(BTNode *b)		//求二叉树b的高度
{
   	int lchildh,rchildh;
   	if (b==NULL) return(0); 				//空树的高度为0
   	else  
	{
		lchildh=BTHeight(b->lchild);	//求左子树的高度为lchildh
		rchildh=BTHeight(b->rchild);	//求右子树的高度为rchildh
		return (lchildh>rchildh)? (lchildh+1):(rchildh+1);
   	}
}*/

void DispBTree(BTNode *b)  //以括号表示法输出二叉树
{
	if (b!=NULL)
	{	printf("%c",b->data);
		if (b->lchild!=NULL || b->rchild!=NULL)
		{	printf("(");						//有孩子结点时才输出(
			DispBTree(b->lchild);				//递归处理左子树
			if (b->rchild!=NULL) printf(",");	//有右孩子结点时才输出,
			DispBTree(b->rchild);				//递归处理右子树
			printf(")");						//有孩子结点时才输出)
		}
	}
}
			
void PreOrder(BTNode *b)  			//先序遍历的递归算法
{
	if (b!=NULL)  
	{	
		printf("%c ",b->data);		//访问根结点
		PreOrder(b->lchild);		//递归访问左子树
		PreOrder(b->rchild);		//递归访问右子树
	}
}
/*
void PreOrder1(BTNode *b)			//先序非递归遍历算法
{
	BTNode *St[MaxSize],*p;
    int top=-1;
    if (b!=NULL) 
    {
        top++;						//根结点进栈
        St[top]=b;
        while (top>-1)				//栈不为空时循环
        {
            p=St[top];				//退栈并访问该结点
            top--;
            printf("%c ",p->data);
            if (p->rchild!=NULL)	//有右孩子,将其进栈
			{
               top++;
               St[top]=p->rchild;
			}
            if (p->lchild!=NULL)	//有左孩子,将其进栈
			{
               top++;
               St[top]=p->lchild;
			}
		}
		printf("\n");
	}
}*/
void InOrder(BTNode *b)   		//中序遍历的递归算法
{
	if (b!=NULL)  
	{	
		InOrder(b->lchild);		//递归访问左子树
		printf("%c ",b->data);	//访问根结点
		InOrder(b->rchild);		//递归访问右子树
	}
}
/*
void InOrder1(BTNode *b)		//中序非递归遍历算法
{
	BTNode *St[MaxSize],*p;
	int top=-1;
	if (b!=NULL)
	{
		p=b;
		while (top>-1 || p!=NULL)	
		{
			while (p!=NULL)		//扫描结点p的所有左下结点并进栈
			{
				top++;
				St[top]=p;
				p=p->lchild;
			}
			if (top>-1)
			{
				p=St[top];		//出栈结点p并访问
				top--;
				printf("%c ",p->data);
				p=p->rchild;
			}
		}
		printf("\n");
	}
}*/
void PostOrder(BTNode *b) 			//后序遍历的递归算法
{
	if (b!=NULL)  
	{	
		PostOrder(b->lchild);		//递归访问左子树
		PostOrder(b->rchild);		//递归访问右子树
		printf("%c ",b->data);		//访问根结点
	}
}
/*
void PostOrder1(BTNode *b)			//后序非递归遍历算法
{
	BTNode *St[MaxSize];
	BTNode *p;
	int top=-1;								//栈指针置初值
	bool flag;
	if (b!=NULL)
	{
		do
		{
			while (b!=NULL)					//将b结点的所有左下结点进栈
			{
				top++;
				St[top]=b;
				b=b->lchild;
			}
			p=NULL;							//p指向当前结点的前一个已访问的结点
			flag=true;						//flag为真表示正在处理栈顶结点
			while (top!=-1 && flag)
			{
				b=St[top];					//取出当前的栈顶元素
				if (b->rchild==p)			//右子树不存在或已被访问,访问之
				{
					printf("%c ",b->data);	//访问b结点
					top--;
					p=b;					//p指向则被访问的结点
				}
				else
				{
					b=b->rchild;			//b指向右子树
					flag=false;				//表示当前不是处理栈顶结点	
				}
			}
		} while (top!=-1);
		printf("\n");
	} 
}
*/
int main()
{
	BTNode *b;
	CreateBTree(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))"); 
	printf("二叉树b:");DispBTree(b);printf("\n");

	printf("先序遍历序列:\n");
	printf("    递归算法:");PreOrder(b);printf("\n");
//	printf("  非递归算法:");PreOrder1(b);
	printf("中序遍历序列:\n");
	printf("    递归算法:");InOrder(b);printf("\n");
//	printf("  非递归算法:");InOrder1(b);
	printf("后序遍历序列:\n");
	printf("    递归算法:");PostOrder(b);printf("\n");
//s	printf("  非递归算法:");PostOrder1(b);
	DestroyBTree(b);
	return 1;
}