SDUT数据结构之二叉树：（先序中序）还原二叉树

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

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

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

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

Sample
Input
9
ABDFGHIEC
FDHGIBEAC
Output
5

#include <stdlib.h>
#include <stdio.h>
struct node
{
    char ch;
    struct node *lchild,*rchild;
};
struct node *creatTree(int n,char s1[],char s2[])
{
   struct node *t;
   int i=0;
   if (n==0)
   {
       return NULL;
   }
   t=(struct node *)malloc(sizeof(struct node));
   t->ch=s1[0];
   for (i=0;i<n;i++)
   {
       if (s1[0]==s2[i])
       {
           t->lchild=creatTree(i,s1+1,s2);
           t->rchild=creatTree(n-i-1,s1+i+1,s2+i+1);
       }
   }
   return t;
}
int depth(struct node *t)
{
    int n,m;
    if (t==NULL)
    {
        return 0;
    }else
    {
        m=depth(t->lchild);
        n=depth(t->rchild);
        if (m>n)
        {
            return (m+1);
        }else return (n+1);
    }
}
int main()
{
    int n;struct node *t;
    char s1[50],s2[50];
    while (~scanf("%d\n%s\n%s",&n,s1,s2))
    {
        t=creatTree(n,s1,s2);
        printf("%d\n",depth(t));
    }
    return 0;
}