当前位置: 代码迷 >> 综合 >> pat 1021. Deepest Root
  详细解决方案

pat 1021. Deepest Root

热度:41   发布时间:2024-01-14 23:18:03.0

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

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:rootfill(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;
}


  相关解决方案