P2661 [NOIP2015 提高组] 信息传递
题意:
n
n
n 个同学玩信息传递,第
i
i
i 个同学只会将信息传递给同学
T
i
T_i
Ti,游戏开始时每个人只知道自己的生日,一轮游戏同学
i
i
i 会把信息传递给同学
T
i
T_i
Ti ,当某人从别人口中得知自己生日时游戏结束,求游戏会进行多少轮。
思路:
带权并查集维护节点到祖先节点的距离
code:
#include<bits/stdc++.h>
#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;
}