CF455C Civilization(树的直径+并查集)

首先求出每个连通子树的直径,注意记在根上,可以有向的合并。
然后就是并查集了。合并两个连通子树,使得新直径最小,考虑新直径如果横跨两块,则最优的显然是把两块的直径中点连起来。
所以合并之后直径的最小值就是max(max(d1,d2),(d1+1)/2+(d2+1)/2+1)

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 300010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,q,h[N],num=0,fa[N],mx[N],d[N],ans;bool vis[N];
struct edge{
    int to,next;
}data[N<<1];
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
inline void merge(int x,int y){
    int xx=find(x),yy=find(y);if(xx>yy) swap(xx,yy);fa[yy]=xx;
}
inline void dp(int x,int Fa){
    vis[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;dp(y,x);
        ans=max(ans,mx[x]+mx[y]+1);mx[x]=max(mx[x],mx[y]+1);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();q=read();
    for(int i=1;i<=n;++i) fa[i]=i;
    while(m--){
        int x=read(),y=read();merge(x,y);
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }for(int i=1;i<=n;++i){
        if(vis[i]) continue;ans=0;dp(i,0);d[i]=ans;
    }while(q--){
        int op=read(),x=read();
        if(op==1) printf("%d\n",d[find(x)]);
        else{
            int y=read();
            int xx=find(x),yy=find(y);if(xx==yy) continue;
            fa[yy]=xx;d[xx]=max(max(d[xx],d[yy]),(d[xx]+1)/2+(d[yy]+1)/2+1);
        }
    }return 0;
}

猜你喜欢

转载自blog.csdn.net/Icefox_zhx/article/details/79944723