【东华大学oj】二叉树：以x为根的子树的深度

二叉树：以x为根的子树的深度

时间限制: 1s

类别: DS：树->中等

问题描述

目的：使用C++模板设计二叉树的抽象数据类型（ADT）。并在此基础上，使用二叉树ADT的基本操作，设计并实现简单应用的算法设计。

内容：（1）请参照链表的ADT模板，设计二叉树的抽象数据类型。（由于该环境目前仅支持单文件的编译，故将所有内容都集中在一个源文件内。在实际的设计中，推荐将抽象类及对应的派生类分别放在单独的头文件中。参考教材、课件，以及网盘中的链表ADT原型文件，自行设计二叉树的ADT。）

注意：二叉树ADT的基本操作的算法设计很多要用到递归的程序设计方法。

（2）ADT的简单应用：使用该ADT设计并实现若干应用二叉树的算法设计。

应用4：要求设计一个递归算法，计算二叉树以元素值为x的结点为根的子树的深度。二叉树的存储结构的建立参见二叉树应用1。

参考函数原型：

//求二叉树中以元素值为x的结点为根的子树的深度 （外壳）

template<class ElemType>

void Get_Height( BinaryTree<ElemType> &T, const ElemType &x, int &depth );   //depth为结果  

//查找二叉树中元素值为x的结点 （递归）

template<class ElemType>

void Get_x_Height( BinaryTreeNode<ElemType> *root, const ElemType &x, int &depth );   

//求二叉树中以元素值为x的结点为根的子树的深度 （递归）

template<class ElemType>

int Get_Height_Cursive( BinaryTreeNode<ElemType> *root);

输入说明

第一行：表示无孩子或指针为空的特殊分隔符

第二行：二叉树的先序序列（结点元素之间以空格分隔）

第三行：元素值x

输出说明

第一行：分两种情况输出

      "No Such Node"  //未找到元素值为x的结点

     子树的深度     //找到元素值为x的结点

#include <iostream>
#include <cstdio>
using namespace std;
typedef struct BNode
{
    string data;
    BNode  *lchild, *rchild;
} BNode,*BTree;
void CreateBTree(BTree &T,string q)
{
    string ch;
    cin>>ch;
    if(ch==q)  T=NULL;
    else
    {
        T=new BNode;
        T->data=ch;
        CreateBTree(T->lchild,q);
        CreateBTree(T->rchild,q);
    }
}
int Get_Depth(BTree T)
{
    int l,r;
    if(T)
    {
        l=Get_Depth(T->lchild);
        r=Get_Depth(T->rchild);
        return l>=r ? (l+1):(r+1);
    }
    else
        return 0;
}
int findDepth(BTree T, string x)
{
    int l,r;
    if(T)
    {
        if(T->data==x)
        {
            return Get_Depth(T);
        }
        else
        {
            l=findDepth(T->lchild, x);
            r=findDepth(T->rchild, x);
            return l>r ? l:r;
        }
    }
    else return 0;
}
int main()
{
    BTree T;
    string q;
    cin>>q;
    CreateBTree(T,q);
    string x;
    cin>>x;
    if(findDepth(T,x)==0)
    {
        cout<<"No Such Node"<<endl;
    }
    else if(findDepth(T,x)!=0)
        cout<<findDepth(T,x)<<endl;
    return 0;
}