L2-006 树的遍历 （25 分）

L2-006 树的遍历 （25 分）

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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

输出样例：

4 1 6 3 5 7 2

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 100000;
int s1[MAXN], s2[MAXN], s3[MAXN];
int n;
void f(int p1,int p2,int q1,int q2,int index)//根据后序和中序构造出树
{
	if(p1>p2 || q1>q2) return ;
	int i = p1 ;
	while(s1[i] != s2[q2]) i++;
	s3[index] = s1[i];
	f(p1, i-1, q1, q1+i-1-p1, 2*index+1);
 	f(i+1, p2, q1+i-p1, q2-1, 2*index+2);
}
int main()
{
	scanf("%d",&n);
	for(int i=0; i!=n; i++){
	  scanf("%d",&s2[i]);
	}
	for(int i=0; i!=n; i++){
	  scanf("%d",&s1[i]);
	}
	f(0,n-1,0,n-1,0);
	int cnt = 0; 
	for(int i=0; i<=100000; i++) {
		if(cnt==n) break;
		if(s3[i]){
			cnt++;
			if(i) printf(" ");
			printf("%d",s3[i]);
		}
	}
	return 0;
}