PAT 甲级 1021 Deepest Root (25 分)

题意：找到能让树的深度最深的结点，若不唯一，升序输出。如果给出的图不连通，输出联通块的个数。

思路：并查集+求树的直径。
求出树最深是多少，然后每个结点dfs一次，如果等于那个最深的深度， 如果等于那个最深的深度就输出。

代码：

include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define pb push_back
typedef pair<int,int> pa;
const int N = 2e4+10;
int pre[N],d[N],u,ans;
int ne[N],e[N],h[N],idx;
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int find(int x)
{
    if(x==pre[x])
        return x;
    return pre[x]=find(pre[x]);
}
void dfs(int x,int pre,int deep)
{
    if(deep>ans)
    {
        ans=deep;u=x;
    }
    for(int i=h[x];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j==pre) continue;
        dfs(j,x,deep+1);
    }
}
int main()
{
    memset(h,-1,sizeof h);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
        pre[i]=i;
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
        int fa=find(a),fb=find(b);
        if(fa!=fb) pre[fa]=fb;
        d[a]++;
    }
    int m=0;
    for(int i=1;i<=n;i++)
        if(pre[i]==i) m++;
    if(m>1)
        printf("Error: %d components\n",m);
    else
    {
        dfs(1,-1,0);
        dfs(u,-1,0);
        int x=ans;
        for(int i=1;i<=n;i++)
        {
            ans=0;
            dfs(i,-1,0);
            if(ans==x)
                cout<<i<<endl;
        }
    }
    return 0;
}