[TCL]浙大PAT 1021. Deepest Root

题意

给出$n$个点$n-1$条边，当只有一个联通块时，求有多少个顶点，当它作为根时形成的树最深

思路

网上很多的解法用并查集判断有几个联通块，然后以每个叶子节点进行DFS，这样的复杂度是$O(n^2)$的
其实只要两遍DFS就好了，顺便把联通块个数也判掉

• 用DFS判断联通块，这个应该很好理解
• 第一次DFS求出最深的叶子节点
• 第二次DFS从这些最深的叶子节点中选一个再进行DFS，再求出一批最深的叶子节点
• 答案是两次DFS结果的并集（第一次求出最长路的一侧，第二次求出另一侧，因为所有不同的最长路必有一个公共交点，可以用反证法证明）

代码

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=a;i<b;i++)
#define debug(a) printf("a =: %d\n",a);
const int INF=0x3f3f3f3f;
const int maxn=1e4+50;
const int Mod=1e9+7;
const double PI=acos(-1);
typedef long long ll;
typedef unsigned int ui;
using namespace std;

vector<int> G[maxn];
int n;
bool vis[maxn];
int depth[maxn];
void dfs(int cur,int dep){
    if (vis[cur]) return;
    vis[cur]= true;
    depth[cur]=dep;
    for(auto i:G[cur]){
        dfs(i,dep+1);
    }
}

void getMax(int &ret,int &pos){
    ret=-1,pos=-1;
    for(int i=1;i<=n;i++) {
        if (depth[i]>ret){
            ret=depth[i];
            pos=i;
        }
    }
}

void getNum(int x,set<int> &v){
    for(int i=1;i<=n;i++) if (depth[i]==x) v.insert(i);
}
int main()
{
    #ifndef ONLINE_JUDGE
        //  freopen("in.txt","r",stdin);
    #endif
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++) G[i].clear();
        for(int i=1,u,v;i<n;i++){
            scanf("%d %d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        mem(vis,0);
        int pt=0;
        for(int i=1;i<=n;i++){
            if(!vis[i]) {
                dfs(i,1);
                pt++;
            }
        }
        if (pt>1){
            printf("Error: %d components\n",pt);
            continue;
        }
        set<int> v;

        int mx,pos;
        getMax(mx,pos);
        getNum(mx,v);
        mem(vis,0);
        dfs(pos,1);
        getMax(mx,pos);
        getNum(mx,v);
        for(auto i:v) printf("%d\n",i);
    }
    return 0;
}

测试数据

9
1 3
3 4
4 5
2 3
2 7
2 6
7 8
6 9