数据结构与算法——7-23 还原二叉树 (25分)

7-23 还原二叉树 (25分)

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

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

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

思路：根据前序遍历和中序遍历来确定左右子树，先找到树根即前序遍历第一位，中序序列的树根前面的字符串是其左子树，后面是右子树。再将左子树和右子树的序列分别作为一棵树进入递归，对其重复找出树根和左右子树的处理，每次记录下当前左右子树的深度，最后返回最大深度。

#include<iostream>
using namespace std;

int dfs(string a,string b,int num){
    int i;
    if(num==0) return 0;
    for(i=0;i<num;i++)
    {
        if(b[i]==a[0])
            break;
    }
    int x=dfs(a.substr(1),b,i)+1;//求左子树的深度
    int y=dfs(a.substr(i+1),b.substr(i+1),num-i-1)+1;//求右子树的深度
    return x>y?x:y;
}
int main()
{
    string pre_order,mid_order;
    int num;
    cin>>num;
    cin>>pre_order>>mid_order;
    int cnt=dfs(pre_order,mid_order,num);
    cout<<cnt<<endl;
}