信息传递（带权并查集求最小环

P2661 [NOIP2015 提高组] 信息传递
题意：
$n$ 个同学玩信息传递，第 $i$ 个同学只会将信息传递给同学 $T_i$，游戏开始时每个人只知道自己的生日，一轮游戏同学 $i$ 会把信息传递给同学 $T_i$ ，当某人从别人口中得知自己生日时游戏结束，求游戏会进行多少轮。
思路：
带权并查集维护节点到祖先节点的距离
code：

#include
#define endl '\n'
#define ll long long
#define ull unsigned long long
#define ld long double
#define all(x) x.begin(), x.end()
#define eps 1e-6
using namespace std;
const int maxn = 2e5 + 9;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
ll n, m;
int f[maxn], dis[maxn];
int ans = inf;
int find(int x){
	if(f[x] != x)
	{
		int last = f[x];
		f[x] = find(f[x]);
		dis[x] += dis[last];
	}
	return f[x];
}
void merge(int x, int y){
	int fx = find(x), fy = find(y);
	if(fx != fy){
		f[fx] = fy;
		dis[fx] = 1 + dis[y] - dis[x];
	}
	else ans = min(ans, dis[x] + dis[y] + 1);
}
void work()
{
	cin >> n;
	for(int i = 1; i <= n; ++i) f[i] = i;
	for(int i = 1; i <= n; ++i){
		int x;cin >> x;
		merge(i, x);
	}
	cout << ans;
}

int main()
{
	ios::sync_with_stdio(0);
//	int TT;cin>>TT;while(TT--)
	work();
	return 0;
}