PAT1021 Deepest Root 最深的根（并查集+遍历树）

传送

给定N个点和N-1条边构成的无向图，若超过一个连通分量，输出连通分量的个数，否则输出能构成的树的最大深度

判断连通分量个数考虑并查集

因为数据很小，考虑暴力枚举每个点为根的情况的深度，取最大值。

 

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<n;i++)
#define INF 0x3f3f3f3f
using namespace std;
const int N=1e4+5;
int n,m,k;
vector<int> g[N],ans;
int pre[N];
int vis[N];
int find(int x){
  return x==pre[x] ? x : pre[x]=find(pre[x]);
}

int build(int root,int depth){
    int res=depth;
    vis[root]=1;
    for(auto p:g[root]){
        if(vis[p]) continue;
        res=max(res,build(p,depth+1));
    }
    return res;
}
int main(){
    int n;
    cin>>n;
    int cnt=n;
    rep(i,1,n+1) pre[i]=i;
    rep(i,0,n-1){
        int a,b;
        cin>>a>>b;
        if(find(a