PAT-ADVANCED1021——Deepest Root

我的PAT-ADVANCED代码仓：https://github.com/617076674/PAT-ADVANCED

原题链接：https://pintia.cn/problem-sets/994805342720868352/problems/994805482919673856

题目描述：

题目翻译：

1021 最深的根

一张无环连通图可以被看作是一棵树。树的高度取决于根节点的选取。现在你需要找出使得这棵树最高的树根。这样的根称作最深的根。

输入格式：

每个输入文件包含一个测试用例。对每个测试用例，第一行有一个正整数N(<= 10000)，代表节点数，这些节点编号为1 ~ N。接下来的N - 1行每行描述了一条连接两节点的边。

输出格式：

对每个测试用例，在一行中输出最深的根。如果该最深的根不唯一，按数字升序每行输出一个根。如果所给的图不是一棵树，输出Error: K components，K代表图中的连通分量。

输入样例1：

5
1 2
1 3
1 4
2 5

输出样例1：

3
4
5

输入样例2：

5
1 3
1 4
2 5
3 4

输出样例2：

Error: 2 components

知识点：图的深度优先遍历、图的广度优先遍历

思路：图的深度优先遍历+图的广度优先遍历

先用图的深度优先遍历判断图是否可以看作一棵树，即图中有多少个连通分量。如果只有1个连通分量，则可以看作一棵树。如果有1个以上的连通分量，就不能看作一棵树。

对图中的每个节点都进行广度优先遍历求其深度。

时间复杂度和空间复杂度均是O(N ^ 2)。

C++代码：

#include<iostream>
#include<vector>
#include<queue>

using namespace std;

int n;	//节点个数
vector<int> graph[10000];	//无向图
bool visited[10000];	//标记数组
int level[10000];	//记录广度优先遍历的层数
int index;

void dfs(int nowVisit);
void bfs(int nowVisit);

int main() {
	scanf("%d", &n);
	fill(level, level + n, 0);
	fill(visited, visited + n, false);
	int v1, v2;
	for(int i = 0; i < n - 1; i++) {
		scanf("%d %d", &v1, &v2);
		graph[v1 - 1].push_back(v2 - 1);
		graph[v2 - 1].push_back(v1 - 1);
	}
	int count = 0;
	for(int i = 0; i < n; i++) {
		if(!visited[i]) {
			dfs(i);
			count++;
		}
	}
	if(count > 1) {
		printf("Error: %d components\n", count);
		return 0;
	}
	for(int i = 0; i < n; i++) {
		fill(visited, visited + n, false);
		if(!visited[i]) {
			index = i;
			bfs(i);
		}
	}
	int max = level[0];
	for(int i = 1; i < n; i++) {
		if(level[i] > max) {
			max = level[i];
		}
	}
	for(int i = 0; i < n; i++) {
		if(level[i] == max) {
			printf("%d\n", i + 1);
		}
	}
	return 0;
}

void dfs(int nowVisit) {
	visited[nowVisit] = true;
	for(int i = 0; i < graph[nowVisit].size(); i++) {
		if(!visited[graph[nowVisit][i]]) {
			dfs(graph[nowVisit][i]);
		}
	}
}

void bfs(int nowVisit) {
	queue<int> q;
	q.push(nowVisit);
	visited[nowVisit] = true;
	while(!q.empty()) {
		int qSize = q.size();
		for(int i = 0; i < qSize; i++) {
			int u = q.front();
			q.pop();
			for(int j = 0; j < graph[u].size(); j++){
				int v = graph[u][j];
				if(!visited[v]){
					q.push(v);
					visited[v] = true;
				}
			}
		}
		level[index]++;
	}
}

C++解题报告：