第十周项目1-二叉树算法库
来源:互联网 发布:最新彩票预测软件 编辑:程序博客网 时间:2024/06/08 15:22
btree.h:- #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 CreateBTNode(BTNode *&b,char *str);
- BTNode *FindNode(BTNode *b,ElemType x);
- BTNode *LchildNode(BTNode *p);
- BTNode *RchildNode(BTNode *p);
- int BTNodeDepth(BTNode *b);
- void DispBTNode(BTNode *b);
- void DestroyBTNode(BTNode *&b);
btree.cpp:- #include "btree.h"
- void CreateBTNode(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')
- {
- 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)
- 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)
- {
- 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 BTNodeDepth(BTNode *b)
- {
- int lchilddep,rchilddep;
- if (b==NULL)
- return(0);
- else
- {
- lchilddep=BTNodeDepth(b->lchild);
- rchilddep=BTNodeDepth(b->rchild);
- return (lchilddep>rchilddep)? (lchilddep+1):(rchilddep+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(")");
- }
- }
- }
- void DestroyBTNode(BTNode *&b)
- {
- if (b!=NULL)
- {
- DestroyBTNode(b->lchild);
- DestroyBTNode(b->rchild);
- free(b);
- }
- }
main.cpp:- #include <stdio.h>
- #include "btree.h"
- int main()
- {
- BTNode *b,*p,*lp,*rp;
- char str[MaxSize];
- char x;
- while(gets(str)!=NULL)
- {
- CreateBTNode(b,str);
- printf("二叉树创建成功:\n");
- DispBTNode(b);
- printf("\n");
-
- printf("请输入查找节点:\n");
- scanf("%c",&x);
- p=FindNode(b,x);
- if (p!=NULL)
- {
- lp=LchildNode(p);
- if (lp!=NULL)
- printf("左孩子为%c ",lp->data);
- else
- printf("无左孩子 ");
- rp=RchildNode(p);
- if (rp!=NULL)
- printf("右孩子为%c",rp->data);
- else
- printf("无右孩子 ");
- }
- else
- printf("未找到!");
- printf("\n");
-
- printf("二叉树b的深度为:%d\n\n",BTNodeDepth(b));
-
- DestroyBTNode(b);
- printf("二叉树释放成功!\n");
-
- fflush(stdin);
- }
- return 0;
- }
运行结果: 0 0