#include<stdio.h>
#include<malloc.h>
#define MaxSize 50
typedef char elemtype;
typedef struct node
{ elemtype data; // 结点的值
struct node *lchild,*rchild; // 左右孩子指针
}BTNode;
void CreateBTNode(BTNode *&b,char *str) /*由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];
}
}
BTNode *FindNode(BTNode *b,elemtype x) /*返回data域为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) /*返回*p结点的左孩子结点指针*/
{
return p->lchild;
}
BTNode *RchildNode(BTNode *p) /*返回*p结点的右孩子结点指针*/
{
return p->rchild;
}
int BTNodeDepth(BTNode *b) /*求二叉树b的深度*/
{
int lchilddep,rchilddep;
if (b==NULL)
return(0); /*空树的高度为0*/
else
{
lchilddep=BTNodeDepth(b->lchild); /*求左子树的高度为lchilddep*/
rchilddep=BTNodeDepth(b->rchild); /*求右子树的高度为rchilddep*/
return (lchilddep>rchilddep)? (lchilddep+1):(rchilddep+1);
}
}
int Nodes(BTNode *b) /*求二叉树b的结点个数*/
{
int num1,num2;
if (b==NULL)
return 0;
else if (b->lchild==NULL && b->rchild==NULL)
return 1;
else
{
num1=Nodes(b->lchild);
num2=Nodes(b->rchild);
return (num1+num2+1);
}
}
void DispBTNode(BTNode *b) /*以括号表示法输出二叉树*/
{
if (b!=NULL)
{
printf("%c",b->data);
if (b->lchild!=NULL || b->rchild!=NULL)
{
printf("(");
DispBTNode(b->lchild);
if (b->rchild!=NULL) printf(",");
DispBTNode(b->rchild);
printf(")");
}
}
}
int count(BTNode *b,int &i) /*求二叉树b的子叶个数*/
{
if(b)
{ if(b->lchild==NULL&&b->rchild==NULL)
i++;
count(b->lchild,i);
count(b->rchild,i);
}
return i;
}
void preorder(BTNode *b)
{
if (b!=NULL)
{
printf("%c",b->data);
preorder(b->lchild);
preorder(b->rchild);
}
}
void inorder(BTNode *b)
{
if (b!=NULL)
{
inorder(b->lchild);
printf("%c",b->data);
inorder(b->rchild);
}
}
void postorder (BTNode *b)
{
if(b!=NULL)
{
postorder(b->lchild);
postorder(b->rchild);
printf("%c",b->data);
}
}
int main ()
{
char a[20],d,h;
int high,number,num,k=0,j;
BTNode *b,*q,*z,*x;
gets(a);
printf("你输入的二叉树是%s\n",a);
CreateBTNode(b,a);
printf("请输入你想查找的节点");
scanf("%C",&d);
q=FindNode(b,d);
printf("左孩子节点为");
z=LchildNode(q);
printf("%c\n",z->data);
printf("右孩子节点为");
x=RchildNode(q);
printf("%c\n",x->data);
high=BTNodeDepth(b);
printf("树的高度是%d\n",high);
number=Nodes(b);
printf("数的总结点数为%d\n",number);
j=count(b,k);
printf("子叶结点个数%d\n",j);
printf("输出的二叉树为");
DispBTNode(b);
printf("\n");
printf("先序输入");
preorder(b);
printf("\n");
printf("中序输入");
inorder(b);
printf("\n");
printf("后序输入");
postorder (b);
printf("\n");
return 0;
}
|