并查集3

并查集之

判断图中的环与大小

信息传递

题目描述

有 n 个同学（编号为 1到 n ）正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象，其中，编号为 i 的同学的信息传递对象是编号为 Ti​ 的同学。

游戏开始时，每人都只知道自己的生日。之后每一轮中，所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象（注意：可能有人可以从若干人那里获取信息， 但是每人只会把信息告诉一个人，即自己的信息传递对象）。当有人从别人口中得知自 己的生日时，游戏结束。请问该游戏一共可以进行几轮？

输入输出格式

输入格式：

共 2 行。

第 1 行包含1个正整数 n ，表示 n 个人。

第 2 行包含 n 个用空格隔开的正整数 T1​,T2​,⋯⋯,Tn​ ，其中第 i个整数 Ti表示编号为 i 的同学的信息传递对象是编号为 Ti的同学

输出格式：

1 个整数，表示游戏一共可以进行多少轮。

输入输出样例

输入样例#1： 

5
2 4 2 3 1

输出样例#1：

3

只要有一个环,自己的信息就会传回来

答案为最小环 

假如说信息由A传递给B，那么就连一条由A指向B的边，同时更新A的父节点

A到它的父节点的路径长也就是B到它的父节点的路径长+1。

如果有两个点祖先节点相同，那么就可以构成一个环，长度为两个点到祖先节点长度之和+1。 

#include<bits/stdc++.h>
#define N 200005
using namespace std;
int fa[N],d[N],ans=0x3fffffff,n;
int read(){
	int cnt=0;char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
int find(int x){
	if(x==fa[x]) return x;
	int root=find(fa[x]);
	d[x]+=d[fa[x]];
	return fa[x]=root;
}
void solve(int x,int y){
	int x1=find(x),y1=find(y);
	if(x1!=y1) fa[x1]=y1,d[x]=d[y]+1;
	else ans=min(ans,d[x]+d[y]+1);
}
int main(){
	n=read();
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=n;i++){
		int t=read();
		solve(i,t);
	}
	cout<<ans;
	return 0;
}

注意这里d的求法,先一直递归到最老的祖先,然后回去时一层层地求d[x],求完再路径压缩