4-14 还原二叉树

4-14 还原二叉树

分数 15
作者 DS课程组
单位 浙江大学

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

输入格式:

输入首先给出正整数N（≤50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)

思路：

利用先序遍历找到根结点，再用中序遍历分割左右子树，最后对左右子树分别递归做相同的操作

代码样例：

//先序遍历 MLR 
//中序遍历 LMR
//先用先序遍历找到根结点，再用中序分割左子树和右子树，再将左子树和右子树逐层递归做相同的操作
#include<bits/stdc++.h>
using namespace std;
typedef char ElementType;
typedef struct Node *BinTree;//将结构体类型命名为BinTree
struct Node//定义结构体类型
{
    ElementType Data;//存数据
    BinTree Left;//左二叉树
    BinTree Right;//右二叉树
};
BinTree CreateTree(char pre[],char mid[],int n);
int GetTreeHigh(BinTree t);
int main()
{
    int N;
    int h;
    char pre[51],mid[51];
    BinTree t;
    cin>>N;
    for(int i=0;i<N;i++)
        cin>>pre[i];
    for(int i=0;i<N;i++)
        cin>>mid[i];
    t=CreateTree(pre,mid,N);
    h=GetTreeHigh(t);
    cout<<h;
    return 0;
}
BinTree CreateTree(char pre[],char mid[],int n)
{
    BinTree t;
    int i=0;
    if(n==0)
        return NULL;
    t=(BinTree)malloc(sizeof(struct Node));
    t->Data=pre[0];
    for(i=0;i<n;i++)//根据中序遍历分割左右子树
        if(mid[i]==pre[0])
            break;
    t->Left=CreateTree(pre+1,mid,i);
    t->Right=CreateTree(pre+i+1,mid+i+1,n-i-1);
    return t;
}
int GetTreeHigh(BinTree t)
{
    if(t==NULL)
        return 0;
    return GetTreeHigh(t->Left)>GetTreeHigh(t->Right)?GetTreeHigh(t->Left)+1:GetTreeHigh(t->Right)+1;
}