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

时间:2022-07-21 20:45:13

给一手链接 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+;
int read(){
int ans=,f=,c=getchar();
while(c<''||c>''){if(c=='-') f=-; c=getchar();}
while(c>=''&&c<=''){ans=ans*+(c-''); 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=;i<=n;i++) f[i]=i,d[i]=;
for(int i=;i<=n;i++){
T=read();
int q=find(T);
if(i==q) ans=min(ans,d[i]+d[T]+);
else f[i]=q,d[i]=d[T]+;
}
printf("%d\n",ans);
return ;
}