算法学习-根据后续和中序遍历输出前序遍历

题目描述

本题要求根据给定的一棵二叉树的后序遍历和中序遍历结果，输出该树的先序遍历结果。

输入描述

第一行给出正整数N(≤30)，是树中结点的个数。随后两行，每行给出N个整数，分别对应后序遍历和中序遍历结果，数字间以空格分隔。题目保证输入正确对应一棵二叉树。

输出描述

在一行中输出Preorder: 以及该树的先序遍历结果。

//分析：
/*
    前序：根左右    中序：左根右    后序：左右根
    由上可看出后序遍历最后一个为根节点，借助该根节点，可以将中序遍历分割成左子树和右子树
    之后通过递归依次删减根节点直到根节点为空
    
    知识点：DFS，string
    
    初次尝试时，使用string的构思是只处理个位数，但是N<=30，存在二位的整数，所以WA了，之后
    看到我对输入的处理，恍然大雾，对于二位整数存入时+‘0’会转化为一个不会重复的字符，输出时
    -‘0’就可以了；
*/
#include<bits/stdc++.h>
using namespace std;

string mid,post;
int n,t;

void Dfs(string mid,string post) {
	if(post.empty())
		return;//递归出口，post为空
	char root=*(post.end()-1);
    //根节点索引为后序遍历最后一个元素
	post.erase(post.end()-1,post.end());
    //删除
	int k=mid.find(root);
    //在中序遍历中找到根节点的位置

	string lmid=mid.substr(0,k);
	string rmid=mid.substr(k+1);
	string lpost=post.substr(0,k);
	string rpost=post.substr(k);
    //将两个字符串分成左右子树

	cout<<root-'0'<<" ";//输出根节点
	Dfs(lmid,lpost);//左子树遍历
	Dfs(rmid,rpost);//右子树遍历
    //构成前序遍历：根左右
}

int main() {
	cin>>n;
    
    //将两个序列转化为string形式
	for(int i=0; i<n; i++)
		cin>>t,post+=t+'0';
	for(int i=0; i<n; i++)
		cin>>t,mid+=t+'0';
    
	cout<<"Preorder: ";
	Dfs(mid,post);
	return 0;
}