【习题 6-11 UVA - 10410】Tree Reconstruction

【链接】 我是链接,点我呀:)

【题意】

在这里输入题意

【题解】

可以先确定当前这棵子树的dfs序的范围。
然后第一个元素肯定是这棵子树的根节点。
那么只要在这棵子树的范围里面枚举节点。
看看有没有下一个bfs序的节点即可。
如果有的话,那么就说明这个根节点有多个子树。
则加入到它的儿子里面去。
然后确定他的每一个儿子的子树的dfs序的范围。
一直往下递归就好。
用bfs模拟,这样可以保证每次进入某个节点的时候,bfs序的下一个节点就是这个子树的第一个儿子节点。

【代码】

/*
1.Shoud it use long long ?
2.Have you ever test several sample(at least therr) yourself?
3.Can you promise that the solution is right? At least,the main ideal
4.use the puts("") or putchar() or printf and such things?
5.init the used array or any value?
6.use error MAX_VALUE?
7.use scanf instead of cin/cout?
8.whatch out the detail input require
*/
#include <bits/stdc++.h>
using namespace std;

const int N = 1e3;

int n, B[N + 5], D[N + 5];
queue < pair<int, int> > dl;
vector <int> G[N + 10];

int main() {
#ifdef LOCAL_DEFINE
	freopen("F:\\c++source\\rush_in.txt", "r", stdin);
#endif
	ios::sync_with_stdio(0), cin.tie(0);
	while (cin >> n) {
		for (int i = 1; i <= n; i++) cin >> B[i];
		for (int i = 1; i <= n; i++) cin >> D[i];
		for (int i = 1; i <= n; i++) G[i].clear();
		dl.push(make_pair(1, n));
		int now = 2;
		while (!dl.empty()) {
			int l = dl.front().first, r = dl.front().second;
			dl.pop();
			int root = D[l];
			if (l >= r) continue;

			G[root].push_back(B[now]);
			now++;
			if (now > n) continue;

			int pre = l + 1;

			for (int i = l + 2; i <= r; i++) {
				if (now <= n && D[i] == B[now]) {
					dl.push(make_pair(pre, i-1));
					pre = i;
					G[root].push_back(D[i]);
					now++;
				}
			}
			dl.push({ pre, r });
		}
		for (int i = 1; i <= n; i++) {
			cout << i << ":";
			for (int x : G[i]) cout << ' ' << x;
			cout << endl;
		}
	}
	return 0;
}