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

最新推荐文章于 2023-12-29 22:41:54 发布

符义的博客

最新推荐文章于 2023-12-29 22:41:54 发布

阅读量370

点赞数

分类专栏：比赛上我会的题…… 文章标签：深度优先哈希算法 c++

本文链接：https://blog.youkuaiyun.com/MortyC99430/article/details/123508341

版权

比赛上我会的题…… 专栏收录该内容

1 篇文章

订阅专栏

这篇博客讨论了如何通过给定的二叉树先序和中序遍历序列来计算二叉树的高度。首先提到了使用递归构建二叉树的方法，然后通过深度优先搜索（DFS）来确定树的高度。后来，作者发现可以通过仅计算左右子树的高度来避免构建整个树，从而简化了问题。示例代码分别展示了两种方法，并给出了测试用例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

输入格式:
输入首先给出正整数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;
}