實現二叉樹各種基本運算的算法

#include<stdio.h>
#include<iostream>
#include<malloc.h>
using namespace std;
#define MaxSize 100
typedef char ElemType;
typedef struct node
{
    ElemType data;//數據類型
    struct node *lchild;//指向左孩子
    struct node *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);
        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(")");
        }
    }
}
/*少了求二叉樹的寬度*/
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);//返回左右樹結點樹加1
    }
}
int LeafNodes(BTNode *b)//求二叉樹b葉子結點個數
{
    int num1,num2;
    if(b==NULL)//空樹的情況
        return 0;
    else if(b->lchild==NULL&&b->rchild==NULL)//為葉子結點的情況
        return 1;
    else//其他情況
    {
        num1=LeafNodes(b->lchild);
        num2=LeafNodes(b->rchild);
        return (num1+num2);//返回左右子樹葉子結點樹
    }
}
int main()
{
    BTNode *b,*p,*lp,*rp;
    CreateBTNode(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))");
    printf("(1)輸出二叉樹:");
    DispBTNode(b);
    printf("\n");
    printf("(2)'H'結點:");
    p=FindNode(b,'H');
    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(" 無右孩子");
    }
    printf("\n");
    printf("(3)二叉樹b的深度:%d\n",BTNodeDepth(b));
    /*少了寬度的輸出*/
    printf("(5)二叉樹b的結點個數:%d\n",Nodes(b));
    printf("(6)二叉樹b的葉子結點個數:%d\n",LeafNodes(b));
    return 0;
}