pat 1021. Deepest Root

这题比较简单，我的思路是

1，从任意点开始一次bfs找到最远的节点，将这些节点都加到vector中，如果bfs不能一次遍历所有的点，计算components输出Error

2，从这些节点的任意一个点开始第二次bfs，找到最远的点，将这些点中没有在vector中的点加到vector中，

3，排序，从小到大输出vector中的点

注意只有一个节点的情况

代码

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

int n,map[10005][10005],isRoot[100005];
int visited[10005],level[10005];
int failed=0,components=1;


int bfs(int root)
{
	//this bfs return the level of tree of root:root
	fill(visited,visited+10005,0);
	fill(level,level+10005,0);
	int max=1;
	queue<int>q;
	q.push(root);
	level[root]=1;
	int current=0;
	for(int i=1;i<=n;i++)
	{
		if(!q.empty())
		{
			while(!q.empty())
			{
				current = q.front();
				q.pop();
				visited[current]=1;
				for(int i=1;i<=n;i++)
				{
					if(map[current][i]==1 && visited[i]==0)
					{
						level[i]=level[current]+1;
						max=level[i]>max?level[i]:max;
						visited[i]=1;
						q.push(i);
					}
				}
				visited[current]=2;//finished visiting
			}
		}
		else
		{
			if(visited[i]==0)
			{
				failed=1;
				q.push(i);
				components++;
			}
		}
	}
	return max;
}

int main()
{
	freopen("F:/pat/data/1021.txt","r",stdin);
	cin>>n;
	int a,b;
	vector <int> vec;
	for(int i=0;i<n-1;i++)
	{
		cin>>a>>b;
		map[a][b]=map[b][a]=1;
	}
	int dep=bfs(1);
	if(failed)
	{
		cout<<"Error: "<<components<<" components"<<endl;
		return 0;
	}
	for(int i=1;i<=n;i++)
	{
		if(level[i]==dep && isRoot[i]==0)
		{
			isRoot[i]=1;
			vec.push_back(i);
		}
	}
	int max_dep=bfs(vec.front());
	for(int i=1;i<=n;i++)
	{
		if(level[i]==max_dep && isRoot[i]==0)
		{
			isRoot[i]=1;
			vec.push_back(i);
		}
	}
	sort(vec.begin(),vec.end());
	for(int i=0;i<vec.size();i++)
		cout<<vec[i]<<endl;
	return 0;
}