洛谷P2661 信息传递(最小环,并查集)

洛谷P2661 信息传递

最小环求解采用并查集求最小环。

只适用于本题的情况。对于新加可以使得两个子树合并的边,总有其中一点为其中一棵子树的根。

复杂度 \(O(n)\) 。

#include<bits/stdc++.h>

using namespace std;

const int inf = 0x3f3f3f3f;
const int maxn = 200005;
int f[maxn], dist[maxn];
int n, to, ans; int father(int x)
{
if(f[x] != x){
int tmp = f[x];
f[x] = father(f[x]);
dist[x] += dist[tmp];
}
return f[x];
}
void solve(int a, int b)
{
int af = father(a), bf = father(b);
if(af == bf){
ans = min(ans, dist[a] + dist[b] + 1);
}
else{
f[af] = bf;
dist[a] = dist[b] + 1;
}
}
int main()
{
scanf("%d", &n);
for(int i = 1; i <= n; i++) f[i] = i;
ans = inf;
for(int i = 1; i <= n; i++){
scanf("%d", &to);
solve(i, to);
}
printf("%d\n", ans);
return 0;
}
上一篇:oracle 基本操作


下一篇:java的子类覆盖梗