二叉树：给定前序数组+中序数组找出后序数组和层序数组

例题：

L2-006 树的遍历

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

输入格式：

输入第一行给出一个正整数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<algorithm>
#include<iostream>
#include<cstring>
#include<vector> 
#include<map>
#include<string>
#include<set>
#include<queue>
#include<set>

typedef long long ll;
using namespace std;

const int N=35;
int a[N],b[N],n,vis=0;//后序，中序数组 

struct node{
	int lc=-1,rc=-1;
}c[N];//树
 
void find(int root,int l,int r){//找出整棵树，root是a[]的下标，l,r是数组b[]的查找范围
	if(l>r)return;
	
	int f=0;
	for(int i=l;i<=r;i++){
		if(a[root]==b[i]){
			f=i;
			break;
		}
	}
	
	if(l<=f-1){
		find(root-(r-f)-1,l,f-1);
		c[root].lc=root-(r-f)-1;
	}
	
	if(f+1<=r){
		find(root-1,f+1,r);
		c[root].rc=root-1;
	}
}

void bfs(){
	queue<int> q;
	
	int flag=0;
	q.push(n);
	while(q.size()>0){
		int t=q.front();
		q.pop();
		
		if(c[t].lc!=-1)q.push(c[t].lc);
		if(c[t].rc!=-1)q.push(c[t].rc);
		
		if(flag)cout<<" ";
		cout<<a[t];
		
		flag=1;
	}
} 

void dfs(int root){
	if(c[root].lc!=-1)dfs(c[root].lc);
	if(c[root].rc!=-1)dfs(c[root].rc);
	
	if(vis)cout<<" ";
	cout<<a[root];
	vis=1;
}
int main(){
	cin>>n;
	
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)cin>>b[i];
	
	find(n,1,n); 

	cout<<"层序遍历:";
	bfs();//层序遍历
	
	cout<<endl;
	
	cout<<"后序遍历:";
	dfs(n);//后序遍历 
	 
	return 0;
}