图论-最小环问题（dfs+并查集）

题目描述

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

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

输入格式

共22行。

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

第22行包含 nn 个用空格隔开的正整数 T_1,T_2,\cdots\cdots,T_nT1​,T2​,⋯⋯,Tn​ ，其中第 ii 个整数 T_iTi​ 表示编号为 ii 的同学的信息传递对象是编号为 T_iTi​ 的同学， T_i \leq nTi​≤n 且 T_i \neq iTi​≠i 。

输出格式

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

输入输出样例

输入 #1复制

5
2 4 2 3 1

输出 #1复制

3

#include <iostream>
#include <algorithm>
#include <cstring>
#include <map>
#include <cstdio>
#include <cmath>
#include <vector>
#include <queue>
using namespace std;

const int inf=1e9;
const int N=200020;
int n,m,step,ans;
int d[N],num[2010],e[N],ne[N],idx,w[N],st[2010],h[N],f[N];
int a[N],vis[N],cnt[N];
typedef pair<int ,int > PII;



/*  搜索做法
void  dfs(int x,int s)
{
    if(f[x]) return;//之前遍历过的点
    if(vis[x])//这一轮遍历的点
    {
        ans=min((s-cnt[x]),ans);
        return ;
    }
    vis[x]=1;
    cnt[x]=s;
    dfs(a[x],s+1);
    f[x]=1;
}

int main(){
    cin>>n;
    ans=inf;
    idx=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;i++){
      dfs(i,1);
    }
    cout<<ans<<endl;
    return 0;
}*/


/*并查集做法*/
int find(int x)
{
    if(x!=f[x])
    {
        int z=f[x];
        f[x]=find(f[x]);
        d[x]=d[x]+d[z];
    }
    return f[x];
}

void Union(int x,int y)
{
    int fa=find(x);
    int fy=find(y);
    if(fa==fy)
        ans=min(ans,d[x]+d[y]+1);//有相同父节点说明本来就在同一条链上，再加上他们之间的边所以形成环，更新最小环
    else
    {
        f[fa]=fy;
        d[x]=d[y]+1;
    }
}
int main()
{
    cin>>n;
    ans=inf;
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        Union(i,a[i]);
    }
    cout<<ans<<endl;
    return 0;
}