二叉樹的建立,查詢,輸出,先序,中序 ,後序遍歷 具體操作
阿新 • • 發佈:2018-12-13
//註釋改日補上。。。
#include<iostream> #include<cstdio> #include<cstdlib> #include<malloc.h> using namespace std; typedef char ElemType; typedef struct node { ElemType data; struct node *lchild; struct node *rchild; }BTNode; void CreatBTree(BTNode *&b, char *str)//建立二叉樹 { BTNode *St[100],*p; 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]; } } 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 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); } } 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 *Lchild(BTNode *b) { return b->lchild; } BTNode *Rchild(BTNode *b) { return b->rchild; } void Destory(BTNode *b) { if(b!=NULL) { Destory(b->lchild); Destory(b->rchild); free(b); } } int main() { BTNode *b,*p,*lp,*rp; char x; CreatBTree(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))"); cout<<"這顆二叉樹是這個樣子的:"<<endl; dispBTree(b); cout<<endl; cout<<"先序遍歷是這樣的"<<endl; PreOrder(b); cout<<endl; cout<<"中序遍歷是這樣的"<<endl; InOrder(b); cout<<endl; cout<<"後序遍歷是這樣的"<<endl; PostOrder(b); cout<<endl; cout<<"請輸入你要查詢的節點資料"<<endl; cin>>x; p=FindNode(b,x); if(p!=NULL) { lp=Lchild(p); if(lp!=NULL) { printf("左孩子 %c\n",lp->data); } else { printf("沒有左孩子"); } rp=Rchild(p); if(rp!=NULL) { printf("右孩子 %c",rp->data); } else { printf("沒有右孩子"); } } Destory(b); return 0; }