PAT 甲级 1021 Deepest Root (25分) 并查集 DFS

原题链接：1021 Deepest Root (25分)

题目大意：

    输入N，给 N-1组点与点连通的数据，让我们判断这是否是一颗树。
    如果是的话进行 DFS，求出这棵树的深度。
    如果不是的话，输出连通块的数量。

分析：

题目要求时间是0.2s，那么2e8的复杂度我们可以勉强通过。
这个题的思路：
首先我们要建树，那就用到了邻接表建树、初始化并查集。然后读入n-1个点与点的关系，如果没连通的话进行连通。在读入的过程中要记录连通块的数量，因为：对于n个点，n-1条边的图，如果只有一个连通块，那就说明它是一棵树。
然后我们分别对每个点进行DFS，遍历以它为父节点能够到达的最深的depth。通过一个 vector记录以以某点为父可以达到最深的点。

满分代码：

#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <algorithm>
#define inf 0x3f3f3f3f
typedef long long LL;
using namespace std;
const int N = 1e4+10, M = 2e4+10;
int n;
int h[N], e[M], ne[M], idx;
int f[N];

int get(int x){
	if(x == f[x])	return x;
	return f[x] = get(f[x]);
}

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

int dfs(int u, int father) {
	int depth = 0;
	// 对以u为根节点的树进行遍历 
	for(int i = h[u]; ~i; i = ne[i]) {
		int j = e[i];
		// 不能沿着father往回找 
		if(j == father)    continue;
		// 不断DFS  max的第二个值 代表u的子结点到最下边的深度加u到子结点的长度 
		depth = max(depth, dfs(j, u) + 1);
	}
	return depth;
}

int main() {
	cin >> n;
	// 注意 必须要初始化邻接表
	memset(h, -1, sizeof h);
	// 并查集f数组 
	for(int i = 1; i <= n; i++)    f[i] = i;
	
	// 刚开始默认k是n个不连通的点 
	int k = n; 
	for(int i = 0; i < n - 1; i++) {
		int a, b;
		cin >> a >> b;
		// 之前未连通 
		if(get(a) != get(b)) {
			k--;
			f[get(a)] = get(b);
		}
		// 让它俩标记连通 
		add(a, b), add(b, a);
	}
	if(k > 1) {
		printf("Error: %d components", k);
	} else {
		// 最深的根的集合 
		vector<int> nodes;
		int max_depth = -1;
		for(int i = 1; i <= n; i++) {
			// 以i为父结点进行DFS 
			int depth = dfs(i, -1); 
			// 更深的情况下将之前的深度清空 
			if(depth > max_depth) {
				max_depth = depth;
				nodes.clear();
				nodes.push_back(i);
			}
			// 如果根的深度相同 
			else if(depth == max_depth) { 
				nodes.push_back(i);
			}
		}
		for(auto v: nodes) {
			cout << v << endl;
		}
	}
	return 0;
}