1021 Deepest Root (25分)

题目大意：给出n个点，n-1条双向边，求这个图是否是一棵树，若是一棵树，按照序号递增输出能够使该树最高的树根，否则输出该图有几个连通块。

思路：

1. 首先通过并查集来判断该图是否连通
2. 若连通，则以1号节点为根进行一次dfs，找到当前树中最深的子节点，那么可以保证这些节点一定存在于结果中。
3. 从第二步得到的节点中任选一个，再进行一次dfs，再找到该树中最深的子节点。这些节点也属于最终的结果中。
4. 排序、去重后输出答案。（这里使用STL中的unique函数只能拿到21分，玄学问题）

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;

const int N=1e4+5;

int n;
vector<int> g[N],book;
int f[N],cnt;
int mdepth,vis[N];

void Init(){
    for(int i=1;i<=n;i++)
        f[i]=i;
    cnt=n;
}

int Find(int a){
    return f[a]==a? a:f[a]=Find(f[a]);
}

void dfs(int u,int depth){
    mdepth=max(depth,mdepth);
    vis[u]=depth;
    for(int i=0;i<(int)g[u].size();i++)
        if(vis[g[u][i]]==0)
            dfs(g[u][i],depth+1);
}

int main(){
    scanf("%d",&n);
    Init();

    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);

        int ru=Find(u),rv=Find(v);
        if(ru!=rv){
            f[ru]=rv;
            cnt--;
        }

        g[u].push_back(v);
        g[v].push_back(u);
    }

    if(cnt!=1) printf("Error: %d components\n",cnt);
    else{
        memset(vis,0,sizeof(vis));
        mdepth=0;
        dfs(1,1);
        for(int i=1;i<=n;i++)
            if(vis[i]==mdepth)
                book.push_back(i);

        memset(vis,0,sizeof(vis));
        mdepth=0;
        dfs(book[0],1);
        for(int i=1;i<=n;i++)
            if(vis[i]==mdepth)
                book.push_back(i);

        sort(book.begin(),book.end());
        for(int i=0;i<(int)book.size();i++)
            if(!i||book[i]!=book[i-1])
                printf("%d\n",book[i]);
    }

    return 0;
}