DFS
似乎找到了PAT的一些规律,如果考的是数据结构里面的东西,就很少有坑。
而如果考的模拟或者简单的题,就有很多边界条件。
使用了两个全局变量,记录每个起点遍历的最大深度和所有顶点的最大深度。
一次性AC,还是比较开心的!

版本1

只有入度为1的点才可能是最深的根节点,已经尽力优化了,但是还是过不了牛客网的同名题。
PAT提交从原来的1100ms提升到了663ms不知道还能怎么优化。

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=1e4+5;
vector<int> G[maxn];
vector<int> p;
int maxd,maxD=-1;
int vis[maxn],in[maxn];
void DFS(int x,int deep){
	vis[x]=1;
	for(int i=0;i<G[x].size();i++){
		if(vis[G[x][i]]==0) {
			vis[G[x][i]]=1;
			if(deep > maxd) {
				maxd = deep;
			}
			DFS(G[x][i],deep+1);
		}
	}
}

int main(){
	int n,u,v;
	scanf("%d",&n);
	memset(in,0,sizeof(in));
	for(int i=0;i<n-1;i++){
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
		in[u]++;
		in[v]++;
	} 
	

	for(int k=1;k<=n;k++){
		memset(vis,0,sizeof(vis));
		maxd=0;
		if(k==1){
			int cnt=0;
			for(int i=1;i<=n;i++){
				if(vis[i]==0){
					DFS(i,1);
					cnt++;
				}
			}
			if(cnt>1) {
				printf("Error: %d components",cnt);
				return 0;
			}
		}else{
			if(in[k]!=1) continue;
			DFS(k,1);
		}
		if(maxd > maxD){
			p.clear();
			p.push_back(k);
			maxD=maxd;
		}else if(maxd == maxD){
			p.push_back(k);
		}
	}
	for(int i=0;i<p.size();i++){
		printf("%d\n",p[i]);
	}

	return 0;
}

版本2

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
int vis[maxn];
vector<int> G[maxn],ans;
int opt=-1;
struct node{
	int val,layer;
};
int bfs(int r){
	int dep=-1;
	queue<node> q;
	node root;
	root.layer = 0;
	root.val = r;
	q.push(root);
	vis[r]=1;
	while(!q.empty()){
		node tmp = q.front();
		q.pop();
		int u = tmp.val;
		for(int i=0;i<G[u].size();i++){
			int v = G[u][i];
			root.val = v;
			root.layer = tmp.layer + 1;
			if(root.layer > dep) dep = root.layer;
			if(vis[v]==0){
				q.push(root);
				vis[v]=1;
			}
		}
	} 
	return dep;
}
int main(){
	int n,u,v,d;
	scanf("%d",&n);
	for(int i=0;i<n-1;i++){
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	int cnt=0;
	for(int i=1;i<=n;i++){
		memset(vis,0,sizeof(vis));
		if(i==1){
			for(int j=1;j<=n;j++){ //求连通分量的个数 
				if(vis[j]==0){  //深搜一遍后,没有标记的点继续搜 
					d = bfs(j);
					cnt++; 
				}
			}
			if(cnt>1){
				printf("Error: %d components",cnt);
				return 0;
			} 
		}else d = bfs(i);
		
		if(d > opt){
			opt = d;
			ans.clear();
			ans.push_back(i);
		}else if(d == opt){
			ans.push_back(i);
			}
	}
	for(auto it: ans){
		printf("%d\n",it);
	}
	return 0;
}