PAT1021 Deepest Root (25)(树的最深节点)

题意：

给出n-1条边，要求判断能否生成一棵树，并求树中所有以它为根时深度最大的节点。

思路：

我一开始做的时候就很莽的直接用dfs搜索，就有一个节点超时，其实简单的思考就是如果以1为根，那么就是当前最深的几个点，他们之间的距离也就是树最大的深度。可以这么理解：从1到i距离最远，那么以i为根，到1的距离已经是最远，那只要从1开始找次深的点即可。实现方法就是先dfs一遍找出最深的几个点，在用其中一个再次dfs，找到的点的集合和前一个集合的并集就是答案。

#include<string>
#include<cstdlib>
#include<vector>
#include<stack>
#include<queue>
#include<utility>
#include<map>
#include<set>
#include<cstdlib>
#include<stdio.h>
#include<algorithm>
#include<functional>
#include<iostream>
#define INF 0x6ffff
using namespace std;

const int maxn = 10010;
vector<int> edge[maxn];
vector<int> temp;
set<int> res;
bool vis[maxn];
int n;
int deepMax;

void dfs(int u, int depth) {
	if (depth > deepMax) {
		deepMax = depth;
		temp.clear();
		temp.push_back(u);
	}
	else if (depth == deepMax) {
		temp.push_back(u);
	}
	vis[u] = false;
	for (int v = 0; v < edge[u].size(); v++) {
		if (vis[edge[u][v]]) {
			dfs(edge[u][v], depth + 1);
		}
	}
}

int main() {
	cin >> n;
	for (int i = 1; i <= n - 1; i++) {
		int u, v;
		cin >> u >> v;
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	int num = 0;
	int s1;
	fill(vis, vis + maxn, true);
	for (int u = 1; u <= n; u++) {
		if (vis[u]) {
			dfs(u, 1);
			if (u == 1) {
				if (temp.size()) s1 = temp[0];
				for (int i = 0; i < temp.size(); i++) {
					//cout << temp[i] << " ";
					res.insert(temp[i]);
				}
			}
			num++;
		}
	}
	if (num > 1) {
		printf("Error: %d components\n", num);
	}
	else {
		temp.clear();
		deepMax = 0;
		fill(vis, vis + maxn, true);
		dfs(s1, 1);
		for (int i = 0; i < temp.size(); i++)
			res.insert(temp[i]);
		for (auto x : res) {
			cout << x << endl;
		}
	}
	return 0;
}