洛谷p2661信息传递题解-CSDN博客

这个题一眼看上去就是用并查集求最小环。

我们可以设两个数组分别是f,d分别表示该点的爸爸和该点到祖先的距离。

当该点的爸爸等于他时，那他肯定就是祖先。

此时信息就肯定传递完了，此时的整个图中（我们可以把每个人都看成一个节点，就成了一个有向图）就出现了一个环，当然一次只能增加一条边。

如果想要游戏尽快完成（因为题目中说只要有一个人形成环就直接退出），就要求出该点形成的环长度，并求出最小值，这个最小值就是结果。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 2000100
using namespace std;
int n, f[maxn], d[maxn];
int find(int x)
{
    if(f[x] != x)
    {
        int fa = find(f[x]);//我们先记录原来要连得点，顺便完成对d数组的求值
    d[x] += d[f[x]];然后加上d数组
    return f[x] = fa;
    }
    else return x;
} 
int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
        f[i] = i;
    int ans = 0x7fffff;
    for(int i = 1; i <= n ; i++)
    {
        int x;
        scanf("%d", &x);
        int fx = find(x);
        int fy = find(i);
        if(fx == fy)
            ans = min(ans, d[i] + d[x] + 1);
        else
        {
            f[i] = x;
            d[i] = 1;//路径压缩
        }
    }
    cout<<ans;
    return 0;
 }