团体程序设计天梯赛 L2-011 玩转二叉树 (25分)

题目链接：

L2-011 玩转二叉树

思路：

先建树再bfs即可，注意逆序

代码：

#include<bits/stdc++.h>

using namespace std;

const int maxn = 2333;
int n, in[maxn], pre[maxn];
vector<int> tree[maxn];
void dfs(int x, int y, int len) {
	if(len < 2) return;
	int root = pre[y];
	int pos = 0;
	for(; in[x + pos] != root; ++pos);
	if(pos != 0) {
		tree[root].push_back(pre[y + 1]);
		dfs(x, y + 1, pos);	
	}
	if(pos !=  len - 1) {
		tree[root].push_back(pre[y + pos + 1]);
		dfs(x + pos + 1, y + pos + 1, len - pos -1);
	}
}
void bfs(int u) {
	queue<int> que;
	que.push(u);
	while(!que.empty()) {
		int now = que.front();
		que.pop();
		if(now != u) cout << ' ';
		cout << now;
		for(int i = tree[now].size() - 1; i >= 0; --i) que.push(tree[now][i]);
	}
}

int main() {
#ifdef MyTest
	freopen("Sakura.txt", "r", stdin);
#endif	
	cin >> n;
	for(int i = 0 ; i < n; i++) cin >> in[i];
	for(int i = 0 ; i < n; i++) cin >> pre[i];
	dfs(0, 0, n);
	bfs(pre[0]);
	return 0;
}