第十周 项目3-利用二叉树遍历思想解决问题（2）

/* 

 *Copyright(c) 2015, 烟台大学计算机学院 

 *All rights reserved. 

 *文件名称：利用二叉树遍历思想解决问题（2）.cpp 

 *作    者：周洁 

 *完成日期：2015年 11月6日 

 *版 本 号： 

 * 

 *问题描述：假设二叉树采用二叉链存储结构存储，分别实现以下算法，并在程序中完成测试： 输出所有叶子节点； 

 *输入描述：若干数据

 *程序输出：所有叶子节点

*/

 

代码：

（1）头文件
#ifndef BTREE_H_INCLUDED

#define BTREE_H_INCLUDED

#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 *FindNode(BTNode *b,ElemType x);     //返回data域为x的节点指针

BTNode *LchildNode(BTNode *p);  //返回*p节点的左孩子节点指针

BTNode *RchildNode(BTNode *p);  //返回*p节点的右孩子节点指针

int BTNodeDepth(BTNode *b); //求二叉树b的深度

void DispBTNode(BTNode *b); //以括号表示法输出二叉树

void DestroyBTNode(BTNode *&b);  //销毁二叉树

#endif // BTREE_H_INCLUDED

 

（2）源文件
#include <stdio.h>

#include <malloc.h>

#include "btreee.h"

void DispLeaf(BTNode *b)

{

    if (b!=NULL)

    {

        if (b->lchild==NULL && b->rchild==NULL)

            printf("%c ",b->data);

        else

        {

            DispLeaf(b->lchild);

            DispLeaf(b->rchild);

        }

    }

}

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];

    }

}

void DestroyBTNode(BTNode *&b)   //销毁二叉树

{

    if (b!=NULL)

    {

        DestroyBTNode(b->lchild);

        DestroyBTNode(b->rchild);

        free(b);

    }

}

int main()

{

    BTNode *b;

    CreateBTNode(b,"A(B(D,E(H(J,K(L,M(,N))))),C(F,G(,I)))");

    printf("二叉树中所有的叶子节点是: ");

    DispLeaf(b);

    printf("\n");

    DestroyBTNode(b);

    return 0;

}

 

运行结果：