还原二叉树

 还原二叉树

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include<stdio.h>


void Input(int A[],int N){
  for(int i = 0; i < N; i++){
    char a;
    scanf("%c",&a);
    //printf("%c",a);
    A[i] = a - 'a';
  }
}
int FindTreeHeight(int pre[],int in[],int N,int Lp,int Li){
  if(N == 0) return 0;
  int i;
  for(i = Li; i < Li + N; i++){
    if(in[i] == pre[Lp]) break;
  }
  int NL = i - Li;
  int NR = Li + N - 1 - i;
  //printf("%d %d %d %d\n",N,i,pre[Lp],in[i]);
  int HL = FindTreeHeight(pre,in,NL,Lp+1,Li);
  int HR = FindTreeHeight(pre,in,NR,Lp+NL+1,Li+NL+1);
  int res = HL > HR ? HL+1:HR+1;
  return res;
}
int main(){
  int N;
  scanf("%d",&N);
  int pre[N];
  int in[N];
  getchar();
  Input(pre,N);
  getchar();
  Input(in,N);
  int Height = FindTreeHeight(pre,in,N,0,0);
  printf("%d",Height);

  return 0;
}