洛谷P2661 信息传递——并查集

给一手链接 https://www.luogu.com.cn/problem/P2661

这道题就是 并查集求最小环

TIPS：压缩路径的时候d[x]=d[fa【x】]+d[x],而不是d[x]=d[fa【x】]+1 因为路径是压缩过的

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
const int M=1e6+7;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+(c-'0'); c=getchar();}
    return ans*f;
}
int n,T,f[M],d[M];
int find(int x){
    int F=f[x];
    if(F==x) return F;
    f[x]=find(F);
    d[x]=d[F]+d[x];
    return f[x];
}
int ans=M;
int main(){
    n=read();
    for(int i=1;i<=n;i++) f[i]=i,d[i]=0;
    for(int i=1;i<=n;i++){
        T=read();
        int q=find(T);
        if(i==q) ans=min(ans,d[i]+d[T]+1);
        else f[i]=q,d[i]=d[T]+1;
    }
    printf("%d\n",ans);
    return 0;
}