团体程序设计天梯赛 L2-006 树的遍历 (25分)

题目链接：

L2-006 树的遍历 (25分)

思路：

先建树再bfs即可

代码：

#include<bits/stdc++.h>

using namespace std;

const int maxn = 2345;
int n, post[maxn], in[maxn];
vector<int> tree[maxn];
void dfs(int x, int y, int len) {
	if(len < 2) return;
	int rt = post[x + len - 1];
	int pos = 0;
	for(; in[y + pos] != rt; ++pos);
	if(pos > 0) {
		tree[rt].push_back(post[x + pos - 1]);
		dfs(x, y, pos);
	}
	if(pos != len - 1) {
		tree[rt].push_back(post[x + len - 2]);
		dfs(x + pos, y + pos + 1, len - pos - 1);	
	}
}
void bfs(int u) {
	queue<int> que;
	que.push(u);
	while(!que.empty()) {
		int now = que.front();
		if(now != u) cout << ' ';
		cout << now;
		que.pop();
		for(int & x : tree[now]) que.push(x);	
	}
}
int main() {
#ifdef MyTest
	freopen("Sakura.txt", "r", stdin);
#endif	
	cin >> n;
	for(int i = 0; i < n; i++) cin >> post[i];
	for(int i = 0; i < n; i++) cin >> in[i];
	dfs(0, 0, n);
	bfs(post[n - 1]);
	return 0;
}