洛谷 P2661 信息传递 删点后DFS

P2661 信息传递

题目描述

有 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

说明

样例1解释

游戏的流程如图所示。当进行完第33 轮游戏后， 44号玩家会听到 22 号玩家告诉他自己的生日，所以答案为 33。当然，第 33 轮游戏后，22号玩家、 33 号玩家都能从自己的消息来源得知自己的生日，同样符合游戏结束的条件。

对于 30\%30%的数据， n ≤ 200n≤200；

对于 60\%60%的数据， n ≤ 2500n≤2500；

对于100\%100%的数据， n ≤ 200000n≤200000。

 

题解：去除不能形成环的点，然后跑dfs找最小环即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<map>
#include<set>
#include<cmath>
#include<cctype>
#include<ctime>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define MOD 998244353
typedef long long ll;
typedef unsigned long long ull;
const int mm=100000;

using namespace std; 

vector<int> G[200005];
int num[200005];
bool vis[200005];
int n,minn=INF,pos;

void init()
{
    queue<int> q;
    for(int i=1;i<=n;i++)
    {
        if(num[i]==0)  //入度为0放入队列
        {
            q.push(i);
            vis[i]=1;
        }
    }
    while(!q.empty())  //删点
    {
        int p=q.front();
        q.pop();
        for(int i=0;i<G[p].size();i++)
        {
            int k=G[p][i];
            num[k]--;
            if(num[k]==0)  /删完上一个点后，当前点入度为零，加入队列
            {
                vis[k]=1;
                q.push(k);
            }
        }
    }
}

void dfs(int sov,int cnt)
{
    if(sov==pos&&cnt)  //回到初始点
    {
        minn=min(minn,cnt);  //更新最小值
        vis[sov]=1;
        return;
    }
    for(int i=0;i<G[sov].size();i++)
    {
        int k=G[sov][i];
        if(!vis[k])
        {
            vis[k]=1;
            dfs(k,cnt+1);
        }
    }
}

int main()
{
    memset(vis,0,sizeof(vis));
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        G[i].push_back(x);
        num[x]++;  //记录入度
    }
    init();
    for(int i=1;i<=n;i++)
    {
        if(!vis[i])
        {
            pos=i;
            dfs(i,0);
        }
    }
    cout<<minn<<endl;
    return 0;
}