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

/* 

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

 *All rights reserved. 

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

 *作    者：周洁 

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

 *版 本 号： 

 * 

 *问题描述：假设二叉树采用二叉链存储结构存储，分别实现以下算法，并在程序中完成测试： 设计一个算法Level(b,x,h)，返回二叉链b中data值为x的节点的层数  

 *输入描述：若干数据

 *程序输出：返回二叉链b中data值为x的节点的层数  

*/

 

代码：

（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 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 Level(BTNode *b,ElemType x,int h)

{

    int l;

    if (b==NULL)

        return 0;

    else if (b->data==x)

        return h;

    else

    {

        l=Level(b->lchild,x,h+1);

        if (l==0)

            return Level(b->rchild,x,h+1);

        else

            return l;

    }

}

int main()

{

    BTNode *b;

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

    printf("值为\'K\'的节点在二叉树中出现在第 %d 层上n",Level(b,'K',1));

    DestroyBTNode(b);

    return 0;

}

 

运行结果：