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

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

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

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

第一想法：
根据前中序构建二叉树、然后求二叉树高度（力扣上有类似的原题）。
难的地方——构建二叉树。

#include<bits/stdc++.h>
using namespace std;
struct node{
    char val;
    node* left=nullptr;
    node* right=nullptr;
    node():val('c'),left(nullptr),right(nullptr){}
};

node *fy(string &q,string &h,int i,int j,int k,int l,unordered_map<char ,int>& hash){
    if(i>j) return nullptr;
    node *t=new node();
    t->val=q[i];
    int index=hash[q[i]];
    int leftsize=index-k;
    t->left=fy(q,h,i+1,i+leftsize,k,k+leftsize-1,hash);
    t->right=fy(q,h,i+leftsize+1,j,k+leftsize+1,l,hash);
    return t;
}
int dfs(node* t){
    if(!t) return 0;
    int le=dfs(t->left);
    int ri=dfs(t->right);
    return max(le,ri)+1;
}
int main(){
    unordered_map<char ,int> hash;
    int cnt;cin>>cnt;
    string q,h;
    cin>>q>>h;
    for(int i=0;i<cnt;i++){
        hash[h[i]]=i;
    }
    node* t=fy(q,h,0,q.size()-1,0,q.size()-1,hash);
    int ans=dfs(t);
    cout<<ans<<endl;
    return 0;
}

后面看了学习通上的提示，只用求栈的高度就行了。

#include<bits/stdc++.h>
using namespace std;


int fy(string &q,string &h,int i,int j,int k,int l,unordered_map<char ,int>& hash){
    if(i>j) return 0;
    int index=hash[q[i]];
    int leftsize=index-k;
    int height1=fy(q,h,i+1,i+leftsize,k,k+leftsize-1,hash);
    int height2=fy(q,h,i+leftsize+1,j,k+leftsize+1,l,hash);
    return max(height1,height2)+1;
}

int main(){
    unordered_map<char ,int> hash;
    int cnt;cin>>cnt;
    string q,h;
    cin>>q>>h;
    for(int i=0;i<cnt;i++){
        hash[h[i]]=i;
    }
    int t=fy(q,h,0,q.size()-1,0,q.size()-1,hash);
    cout<<t<<endl;
    return 0;
}