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

Code:
无须构造二叉树进行广度优先搜索
·使用板子：后序+中序输出层序，比后序+中序输出先序增加一个变量index表示根结点在二叉树中的下标
·若二叉树的结点按层序遍历顺序存在数组中(根结点下标从0开始)，已知根结点下标为i，则以当前根结点为子树的左右孩子下标分别为2*i+1和2*i+2
· map根据键值从小到大自动排序，递归完成后ans中的值就是层序遍历的顺序

#include<bits/stdc++.h>
using namespace std;
vector<int> post,in;
map<int,int> ans;
void level(int root,int start,int end,int index){
	if(start>end) return;
	int k;
	for(k=start;k<end;k++) if(in[k]==post[root]) break;
	ans[index] = post[root];
	level(root-1-end+k,start,k-1,2*index+1);
	level(root-1,k+1,end,2*index+2);
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int N,flag=0;
	cin >> N;
	post.resize(N);
	in.resize(N);
	for(int i=0;i<N;i++) cin >> post[i];
	for(int i=0;i<N;i++) cin >> in[i];
	level(N-1,0,N-1,0);
	for(int i=0;i<ans.size();i++){
		if(ans[i]){
			if(!flag) flag = 1;
			else cout << " ";
			cout << ans[i];
		}	
	}
	return 0;
}