D - 数据结构实验之二叉树四：（先序中序）还原二叉树

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

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

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

Sample
Input
9
ABDFGHIEC
FDHGIBEAC
Output
5

#include <iostream>
#include <bits/stdc++.h>

using namespace std;

char pre[55],mid[55];
int len;
typedef struct node
{
    char data;
    node *l,*r;
}Tree;

//根据先序遍历、中序遍历还原二叉树
Tree* buildtree(int len,char pre[],char mid[])
{
    Tree *root;
    if(!len)
        return NULL;
    root = new Tree;
    root->data=pre[0];
    int i;
    for(i=0;i<len;i++)
    {
        if(mid[i]==pre[0])
            break;
    }
    root->l=buildtree(i,pre+1,mid);
    root->r=buildtree(len-i-1,pre+i+1,mid+i+1);
    return root;
}

int Deep_tree(Tree *root)
{
    //后序遍历
    int deepl,deepr,Deep=0;
    if(!root)
        return Deep=0;
    else
    {
        deepl=Deep_tree(root->l);
        deepr=Deep_tree(root->r);
        Deep=1+(deepl > deepr ? deepl:deepr);
    }
    return Deep;
}

int main()
{
    ios::sync_with_stdio(false);

    while(cin>>len)
    {
        cin>>pre;
        cin>>mid;
        Tree *root=buildtree(len,pre,mid);
        cout<<Deep_tree(root)<<endl;
    }
    return 0;
}