PAT A1021

最新推荐文章于 2021-05-16 23:14:14 发布

Focus9

最新推荐文章于 2021-05-16 23:14:14 发布

阅读量381

点赞数

分类专栏： PAT 甲级文章标签： PAT 甲级

本文链接：https://blog.csdn.net/weixin_37565706/article/details/78991237

版权

PAT 甲级专栏收录该内容

130 篇文章 2 订阅

订阅专栏

方法：广度遍历，若图不是连通图，则返回连通分量的个数

注意点：图的广度遍历需要vis数组，且入队后即修改vis数组值，树可不用

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn=10010;
vector<int> arc[maxn];
bool vis[maxn];
int N,depth[maxn],maxdepth=-1,layer[maxn];
int BFS(int root){
	queue<int> q;
	fill(layer,layer+maxn,0);
	q.push(root);
	vis[root]=true;
	layer[root]=1;
	int tdepth=0,count=0;
	while(!q.empty()){
		int t=q.front();
		q.pop();
		count++;
		int len=arc[t].size();
		tdepth=max(tdepth,layer[t]);
		for(int i=0;i<len;i++){
			if(vis[arc[t][i]]==false){
				q.push(arc[t][i]);
				layer[arc[t][i]]=layer[t]+1;
				vis[arc[t][i]]=true;
			}
		}
	}
	if(count<N) return -1;
	else return tdepth;
}
int calcomponent(int root){
	fill(vis,vis+maxn,0);
	int count=0;
	for(int i=1;i<=N;i++){
		if(vis[i]==false){
			BFS(i);
			count++;
		}
	}
	return count;
}
int main(){
	scanf("%d",&N);
	for(int i=1;i<N;i++){
		int v1,v2;
		scanf("%d%d",&v1,&v2);
		arc[v1].push_back(v2);
		arc[v2].push_back(v1);
	}
	for(int i=1;i<=N;i++){
		fill(vis,vis+maxn,0);
		depth[i]=BFS(i);
		if(depth[i]==-1){
			printf("Error: %d components",calcomponent(i));    //计算连通分量的个数 
			return 0;
		}
		maxdepth=max(depth[i],maxdepth);
	}
	for(int i=1;i<=N;i++){
		if(depth[i]==maxdepth){
			printf("%d\n",i);
		}
	}
	return 0;
}