【PTA 7-14】根据后序和中序遍历输出先序遍历

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Xylon_/article/details/87783425

                  7-14 根据后序和中序遍历输出先序遍历 （25 分）

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

输入格式:

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

输出格式:

在一行中输出Preorder:以及该树的先序遍历结果。数字间有1个空格，行末不得有多余空格。

输入样例:

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例:

Preorder: 4 1 3 2 6 5 7

推荐阅读：已知后序与中序求前序(先序)序列及代码模板

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))

int post[35],mid[35];

void pre(int root,int start,int end)
{
	if(start>end)
		return ;
	int i=start;
	while(i<end&&mid[i]!=post[root])
		i++;
	cout<<" "<<post[root];
	pre(root-1-end+i,start,i-1);
	pre(root-1,i+1,end);
}

int main()
{
	int n,i;
	cin>>n;
	for(i=0;i<n;i++)
		cin>>post[i];
	for(i=0;i<n;i++)
		cin>>mid[i];
	cout<<"Preorder:";
	pre(n-1,0,n-1);
	return 0;
}