P2661 信息传递

传送门（洛谷）

并查集求最小环

对于两个点$A,B$,如果$A指向B$，则将$A，B$两点连一条边，且将两点放入一个集合中，更新权值，每条边的初始权值就是1，就是步数

inline int find(int x) {
	if(fa[x]!=x) {
		int last=fa[x];
		fa[x]=find(fa[x]);
		dis[x]+=dis[last];
	}
	return fa[x];
}
void check(int x,int y) {
	int fx=find(x),fy=find(y);
	if(fx!=fy)  {
		fa[fx]=fy;
		dis[x]=dis[y]+1;//x指向y，如果不在一个集合，则现在x到祖先的距离就是y到祖先的距离+1
	}
	else minn=min(minn,dis[x]+dis[y]+1);
}

$dis[x]$表示$x$到他的祖先的步数
题中说过，每人只会把信息告诉一个人，那么不可能从一个点同时指向两个点，那么我们就一定可以得出，如果构成环，则一定是这条链的终点，可以画个图
这种情况一个点引出了两个点，是不可能出现的，所以正确图解：

只会出现在终点返回构成环

Code

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int (i)=(a);(i)<=(b);(i)++)
using namespace std;
const int maxn=1e6+10;
const int maxm=3e3+10;
int n,minn=INT_MAX;
int a[maxn],fa[maxn],dis[maxn];

template <class t> inline void read(t &x) {
	x=0;int f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
	x*=f;
}

inline int find(int x) {
	if(fa[x]!=x) {
		int last=fa[x];
		fa[x]=find(fa[x]);
		dis[x]+=dis[last];
	}
	return fa[x];
}

void check(int x,int y) {
	int fx=find(x),fy=find(y);
	if(fx!=fy)  {
		fa[fx]=fy;
		dis[x]=dis[y]+1;
	}
	else minn=min(minn,dis[x]+dis[y]+1);
}

void readdata() {
	read(n);
	rep(i,1,n) {
		fa[i]=i;
		dis[i]=0;
	}
}

void work() {
	rep(i,1,n) {
		read(a[i]);
		check(i,a[i]);
	}
	printf("%d\n",minn);
}

int main() {
	freopen("input.txt","r",stdin);
	readdata();
	work();
	return 0;
}