数据结构实验之二叉树四：还原二叉树

数据结构实验之二叉树四：还原二叉树

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

输入

输入数据有多组，每组数据第一行输入 1 个正整数 N(1 <= N <= 50) 为树中结点总数，随后 2 行先后给出先序和中序遍历序列，均是长度为 N 的不包含重复英文字母 ( 区分大小写 ) 的字符串。

输出

  输出一个整数，即该二叉树的高度。

示例输入

9 
ABDFGHIEC
FDHGIBEAC

示例输出

5

提示

 

来源

 xam

示例程序

#include<bits/stdc++.h>
using namespace std;
char s1[100],s2[100],ans[100];
typedef struct BiTNode
{
    char data;
    struct BiTNode *lchild,*rchild;
} BiTNode,*BiTree;
int i;
void build(BiTree &T,char *s1,char *s2,int n)//建立二叉树
{
    if(n<=0) T=NULL;
    else
    {
        int p=strchr(s2,s1[0])-s2;
        T=new BiTNode;
        T->data=s1[0];
        build(T->lchild,s1+1,s2,p);//递归创建左子树
        build(T->rchild,s1+p+1,s2+p+1,n-p-1);//递归创建右子树
    }
}
int  Depth(BiTree &p)//求深度
{    if (p == NULL)
        return 0;
    else
        return 1 + Depth(p->lchild)>Depth(p->rchild)?
               (Depth(p->lchild)+1):(Depth(p->rchild)+1);
               }
int main()
{
    BiTree T;
    int len;
    while(cin>>len)
    {
        for (int j=0; j<len; ++j)
            cin>>s1[j];
        for (int j=0; j<len; ++j)
            cin>>s2[j];
        build(T,s1,s2,len);
        i=Depth(T);
        cout<<i<<endl;
    }
    return 0;
}