题目链接:https://www.patest.cn/contests/pat-a-practise/1021
题目大意:找出树中能使树的深度达到最大的根。
解题思路:
- 定义深度优先DFS(int v,int high),该函数用来找离v点最远点的集合
- 先任取一点v,进行dfs找到离v点最远的一部分点即为集合set1,这些点是最终要求的点的子集。
- 从set1中任选一点,在进行一次DFS,得到set2。
- set1和set2的并集即为最终所求。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> node[10001];//二维的vector,存储每个结点相连的结点的下标
int visit[10001]={0};
int n;
int maxhigh=0;//最大高度
vector<int> tmp;//存储距离dfs函数中的v点最深的点
void dfs(int v,int high){
if(high>maxhigh){//找到更深的
maxhigh=high;
tmp.clear();//将之前的清除
tmp.push_back(v);
}
else if(high==maxhigh){//距离相等的
tmp.push_back(v);//直接保存
}
visit[v]=1;//标记为已访问
for(int i=0;i<node[v].size();i++){
if(visit[node[v][i]]==0){
dfs(node[v][i],high+1);//递归地访问与该节点临接的顶点,深度加一
}
}
}
int main(int argc, char const *argv[])
{
cin>>n;
int a,b;
int flag[10001]={0};//falg[i]=1表示结点i是最深的根节点之一
for(int i=1;i<=n-1;i++){//输入n-1条边
cin>>a>>b;
node[a].push_back(b);
node[b].push_back(a);
}
int cnt=0;//连通分量的个数
for(int i=1;i<=n;i++){
if(visit[i]==0){
dfs(i,1);//第一次dfs
cnt++;
}
}
if(cnt>1)
cout<<"Error: "<<cnt<<" components"<<endl;
else{
for(int i=0;i<tmp.size();i++){
flag[tmp[i]]=1;
}
int start=tmp[0];//该点为距离1最远的点。
tmp.clear();
for(int i=1;i<=n;i++)
visit[i]=0;
dfs(start,1);//第二次dfs找距离start最远的点
tmp.push_back(start);
for(int i=0;i<tmp.size();i++)
flag[tmp[i]]=1;
for(int i=1;i<=n;i++){
if(flag[i]==1)
cout<<i<<endl;
}
}
return 0;
}