二叉树的各种操作总汇（c/c++语言实现）

本文总结二叉树的各种操作与Ｃ语言的实现

头文件

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

数据结构

typedef struct Node{
	char data ;
	struct Node * rchild ;
	struct Node * lchild ;
}Node,*pNode;

随机创建二叉树的数据

void CreateRandChar(char *const a,int len){ /*随机生成一个长度为len的字符串*/
	int i;
	srand((unsigned int)time(NULL));
	for(i=0;len--;)
		a[i++]='A'+(int)(1.0*rand()/(RAND_MAX)*26);
	a[i]=0;
	return ;
}

创建二叉树

#define  LCHILD(i)		(2*i+1)
#define  RCHILD(i)		(2*i+2)
pNode CreateTree(char * a,int len){ /*创建一棵二叉树*/
	int i = 0,r,l;
	pNode p;
	p = (Node*)malloc(len*sizeof(Node));
	if(!p)
		return NULL ;
	for (i=0;i<len;i++){
		p[i].data = a[i];
		r = RCHILD(i);
		l = LCHILD(i);
		if(l<len)
			p[i].lchild = &p[l] ;
		else
			p[i].lchild = NULL ;
		if(r<len)
			p[i].rchild = &p[r] ;
		else
			p[i].rchild = NULL; 
	}
	return p ;
}

二叉树访问函数

typedef void (*pVisit)(void *arg) ;
void VisitFun (void *arg)/*访问函数*/{
	pNode  p = (pNode)arg ;
	printf("%c,",p->data);	
}

三种递归遍历（前、中、后序）算法

void PreOrderTraverse(pNode head ,pVisit fun){/*递归先序遍历*/
	if(!head)
		return ;
	fun((void*)head);
	if(head->lchild)
		PreOrderTraverse(head->lchild,fun);
	if(head->rchild)
		PreOrderTraverse(head->rchild,fun);
}
void InOrderTraverse(pNode head ,pVisit fun){/*递归中序遍历*/
	if(!head)
		return ;
	if(head->lchild)
		PreOrderTraverse(head->lchild,fun);
	fun((void*)head);
	if(head->rchild)
		PreOrderTraverse(head->rchild,fun);
}
void AfterOrderTraverse(pNode head ,pVisit fun){ /*递归中序遍历*/
	if(!head)
		return ;
	if(head->lchild)
		PreOrderTraverse(head->lchild,fun);
	if(head->rchild)
		PreOrderTraverse(head->rchild,fun);
	fun((void*)head);
}

遍历二叉树的某一层

int OneLevelTraverse(pNode head,int level ,pVisit fun) {/*递归遍历某一层*/
	if(!head || level <=0)
		return 0;
	if( 1==level){
		fun(head);
		return 1 ;
	}
	return OneLevelTraverse(head->lchild,level-1,fun )
		+ OneLevelTraverse(head->rchild,level-1,fun) ;
}

获得二叉树的深度

int GetDepth(pNode head){
	int rd,ld;
	if(!head)
		return 0;
	rd = GetDepth(head->rchild) ;
	ld = GetDepth(head->lchild) ;
	return rd>ld?(rd+1):(ld+1)  ;
}

层次遍历二叉树

/*方法一：递归法层次遍历二叉树*/
void LevelTraverse1(pNode head, pVisit fun ){
	int i,depth ;
	depth = GetDepth(head) ;
	for(i=1;i<=depth;i++)
		OneLevelTraverse(head,i,fun) ;
}

/*方法二：递归法层次遍历二叉树*/
void LevelTraverse2(pNode head, pVisit fun ){
	int i = 0;
	for(i=1;;i++)//当访问到最后一层的下一层的时候，会返回0,退遍历结束
		if(!OneLevelTraverse(head,i,fun))
			break;
}

/*方法三：非递归方式（借用队列）层次遍历二叉树*/
void LevelTraverse3(pNode head, pVisit fun ){
	queue<pNode > que ;
	pNode p ;
	que.push(head) ;
	while(!que.empty()){
		p = que.front();
		fun(p);
		if(p->lchild)	que.push(p->lchild);
		if(p->rchild)	que.push(p->rchild);
		que.pop();
	}
}

/*如果是逆序层次遍历的话，则可以用栈的功能,先全部进栈，则出栈即可*/

main函数测试

int main(int argc,char** argv){
	char a[255];
	int len ;
	pNode head ;
	len = 10;
	CreateRandChar(a,len);
	printf("%s\n",a);
	head= CreateTree(a,len);
	PreOrderTraverse(head,&VisitFun);
	printf("\nlevel traverse:");
	LevelTraverse3(head,&VisitFun) ;
	return 0;
}