树的遍历（中序遍历+前序/后序遍历求原树）

树的遍历有四种：前序遍历、中序遍历、后序遍历、层序遍历。

其中：中序遍历 + 其他任意一种可以唯一确定原树。

一、中序遍历+后序遍历求原树

题目：1497. 树的遍历 - AcWing题库

如图：

#include<bits/stdc++.h>
using namespace std;
const int N = 1e3 + 7;
#define ll long long

int n;
int mid[N],back[N];
map<int,int>l,r,idx;

/* bl,br是当前树在back[]数组中（后序遍历）左右下标
   ml,mr是当前树在mid[]数组中（中序遍历）左右下标*/
int dfs(int ml,int mr,int bl,int br) {
	int root = back[br];//后序遍历最后一个值是根节点
	int k = idx[root];//查找根节点在中序遍历中的下标
	//如果左子树存在，找到根节点左子树值
	if(ml<k)l[root] = dfs(ml,k-1,bl,bl+k-1-ml);//根节点左儿子=左子树
	//如果右子树存在，找到根节点右子树值
	if(mr>k)r[root] = dfs(k+1,mr,bl+k-ml,br-1);//根结点右儿子=右子树
	return root;
	}

//输出层序遍历结果
void bfs(int root) {
    queue<int> q;
    q.push(root);
    vector<int>s;
    while(!q.empty()){
        int t = q.front();
        q.pop();
        s.push_back(t);
        if(l[t]) q.push(l[t]);//如果当前节点存在左子树
        if(r[t]) q.push(r[t]);//如果当前节点存在右子树
    }
    for(int i = 0; i < s.size(); i++) {
    	cout << s[i];
    }
}

int main() {
	cin >> n;
	for(int i = 1;i <= n; i++)
		cin >> back[i]; //存储后序遍历结果
	for(int i = 1;i <= n; i++) {
		cin >> mid[i]; //存储中序遍历结果
		idx[mid[i]] = i; //存储中序遍历下标
}
	int root = dfs(1,n,1,n);
	bfs(root);

	return 0;
}

二、中序遍历+前序遍历求原树

同理只是前序遍历根在前面

#include<bits.stdc++.h>
using namespace std;
const int N=30;

int n;
char preorder[N],inorder[N];
map<char,int> l,r,pos;

char build(int inl,int inr,int prel,int prer)
{
    char root=preorder[prel];
    int k=pos[root];
    if(inl<=k-1) l[root]=build(inl,k-1,prel+1,prel+1+k-1-inl);
    if(k+1<=inr) r[root]=build(k+1,inr,prel+1+k-1-inl+1,prer);
    return root;
}

void print(char root)
{
    if(l.count(root)) print(l[root]);
    if(r.count(root)) print(r[root]);
    cout<<root;
}
int main()
{
    while(scanf("%s",preorder+1)!=EOF){
        l.clear(),r.clear(),pos.clear();
        scanf("%s",inorder+1);
        n=strlen(inorder+1);

        for(int i=1;i<=n;i++) pos[inorder[i]]=i;

        char root=build(1,n,1,n);

        print(root);
        puts("");
    }
    return 0;
}