1020 Tree Traversals

1020 Tree Traversals

题目大意

给出后序和中序序列，打印出广度优先遍历序列

算法思想

• 本来是打算构建二叉树，然后老老实实广度优先，但是构建树的时候，老是出错，看完柳神的代码后，再一次五体投地，太牛了。
• 可以直接由后序和中序找到先序序列，而有了先序序列，可以构建出其顺序存储的二叉树，但是此题不需要，直接挨着存，就是广度优先！！！柳神yyds！！！

代码

#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
vector<int>post,in;
map<int, int>t;
void pre(int root, int start, int end, int wb)//中序和后序转先序
{
	if (start > end) return;
	int id = find(in.begin(), in.end(), post[root]) - in.begin();//找到当前根在中序的位置
	t[wb] = post[root];//赋给map
	pre(root - end + id - 1, start, id - 1, wb * 2 + 1);//左子树，左子树仍然可以由中序和后序推出其各个元素
	pre(root - 1, id + 1, end, wb * 2 + 2);//右子树
	return;
}
int main()
{
	int n;
	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];
	pre(n - 1, 0, n - 1, 1);
	auto it = t.begin();
	cout << it->second;
	for (it++; it != t.end(); it++)//打印需要
		cout << " " << it->second;
	return 0;
}