习题6-11　树重建（Tree Reconstruction, UVa 10410）

最新推荐文章于 2022-11-28 11:37:52 发布

Chen丶HC

最新推荐文章于 2022-11-28 11:37:52 发布

阅读量328

点赞数

分类专栏：紫书第六章习题文章标签： acm uva 算法竞赛入门经典 c++

本文链接：https://blog.csdn.net/chcnsn/article/details/78172142

版权

紫书第六章习题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

看了有一晚了。。也看了别人的题解。。还是感觉不太清楚。。。。

先占个坑，下边是一种做法。。还有根据bfs来分割dfs的做法。。。

update 10.9
去学习了下树的栈遍历，回来感觉能明白这种做法了。
题意即为确定各节点的父子关系，从dfs序中，很显然的，一路下去都是前驱节点的孩子。
只是如何判断此时这一子树已经访问结束呢？
很显然从bfs序中来寻找答案，bfs[u] + 1 < bfs[v]即为子节点。
那么等于1的情况呢？
bfs[u] + 1 == bfs[v]时，u为root必然成立，否则当v < u时可以当做成立。

#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>
#include <sstream>
#include <utility>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <cctype>
#define CLEAR(a, b) memset(a, b, sizeof(a))
#define IN() freopen("in.txt", "r", stdin)
#define OUT() freopen("out.txt", "w", stdout)
#define LL long long
#define maxn 10005
#define maxm 100005
#define mod 1000000007
#define INF 1000000007
#define eps 1e-5
#define PI 3.1415926535898
#define N 26
using namespace std;
//-------------------------CHC------------------------------//
vector<int> son[maxn];
int bfs[maxn];

int main() {
	int n;
	while (~scanf("%d", &n)) {
		for (int i = 1, x; i <= n; ++i)
			scanf("%d", &x), bfs[x] = i, son[i].clear();
		int root;
		scanf("%d", &root);
		stack<int> sta;
		sta.push(root);
		for (int i = 1, x; i < n; ++i) {
			scanf("%d", &x);
			for (;;) {
				int u = sta.top();
				if (bfs[u] + 1 < bfs[x] || u == root) {
					son[u].push_back(x);
					sta.push(x);
					break;
				}
				else sta.pop();
			}
		}
		for (int i = 1; i <= n; ++i) {
			printf("%d:", i);
			for (int j = 0, sz = son[i].size(); j < sz; ++j)
				printf(" %d", son[i][j]);
			puts("");
		}
	}
	return 0;
}