计算二叉树的深度

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include<iostream>
#include<stdlib.h>
int flag=0;
using namespace std;
typedef struct node
{
    struct node *left;
    struct node *right;
    char data;
}Node;
Node *binarytree(char a[],char b[],int len)
{
    int i;
    if (len==0)
    return NULL;
    Node *p;
    p=(Node *)malloc(sizeof(Node));
    p->data=a[0];
    for (i=0;i<len;i++)
    {
        if (b[i]==a[0])
        break;
    }
    p->left=binarytree(a+1,b,i);
    p->right=binarytree(a+i+1,b+i+1,len-1-i);
    return p;
}
用一个judge函数来判断他的深度，当深度大于当前找到的最大深度时，更改最大深度。
void judge(Node *p,int level)
{
    if (p==NULL)
    return ;
    if (level>flag)
    flag=level;
    if (p->left!=NULL)
    judge(p->left,level+1);
    if (p->right!=NULL)
    judge(p->right,level+1);
}
int main()
{
    int i,n;
    cin>>n;
    char pre[n],mid[n];
    for (i=0;i<n;i++)
    cin>>pre[i];
    for (i=0;i<n;i++)
    cin>>mid[i];
    Node *tree;
    tree=(Node *)malloc(sizeof(Node));
    tree=binarytree(pre,mid,n);
    judge(tree,1);
    cout<<flag;
}