NC16466 [NOIP2015]信息传递

题目链接

题目

题目描述

有 n 个同学(编号为 1 到 n)正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为 i 的同学的信息传递对象是编号为Ti的同学。

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

输入描述

第 1 行包含 1 个正整数 n,表示 n 个人。第 2 行包含 n 个用空格隔开的正整数T1,T2, … … ,Tn,其中第 i 个整数Ti表示编号为 i 的同学的信息传递对象是编号为Ti的同学,Ti≤ n 且Ti≠ i。数据保证游戏一定会结束。

输出描述

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

示例1

输入

5
2 4 2 3 1

输出

3

说明

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

备注

对于30%的数据,n≤ 200;
对于60%的数据,n≤ 2500;
对于100%的数据,n≤ 200000。

题解

知识点:并查集,图论。

这道题实际求一个最小环,用dfs也能解决,这里用并查集。

每次增加一条单向边,起点 \(i\) 一定是出度为 \(0\) 的点,即一个集合的祖先。因此,我们用并查集维护一个集合每个点到祖先的距离(链长)。

对于合并 \((i,j)\) ,如果两个点在同一个集合,则 \(j\) 的链长加 \(1\) 即环大小,取所有环最小值即可;如果不在一个集合,则祖先点 \(i\) 的权值应修改为,\(j\) 的权值加 \(1\) ,因为 \((i,j)\) 也是一条边。

路径压缩照常即可。

时间复杂度 \(O(n \log n)\)

空间复杂度 \(O(n)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

struct DSU {
    vector<int> fa, Size;

    explicit DSU(int n) :fa(n + 1), Size(n + 1) {
        for (int i = 1;i <= n;i++)
            fa[i] = i;
    }

    void init(int n) {
        for (int i = 1;i <= n;i++)
            fa[i] = i, Size[i] = 0;
    }

    int find(int x) {
        if (fa[x] == x) return x;
        int tmp = fa[x];
        fa[x] = find(fa[x]);
        Size[x] += Size[tmp];
        return fa[x];
    }

    bool same(int x, int y) { return find(x) == find(y); }

    void merge(int x, int y) {
        int a = find(x), b = find(y);
        if (a == b) return;
        Size[x] = Size[y] + 1;//x本身肯定是根节点(结论),指向另一个集合的某点,那么权值则是指向的那个点权值加一
        fa[a] = b;
    }
};

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    DSU dsu(n);//建立节点指向根节点的带方向的集合,根节点是传递信息的终点,节点权值为到根节点的距离
    int ans = n;
    for (int i = 1;i <= n;i++) {
        int j;
        cin >> j;//注意,i一定是某个集合的根节点,因为出度为0
        if (dsu.same(i, j)) ans = min(ans, dsu.Size[j] + 1);//集合根节点指向集合自身的某个点,则出现环,答案为j到根节点的距离加一
        dsu.merge(i, j);
    }
    cout << ans << '\n';
    return 0;
}
posted @ 2023-01-05 21:13  空白菌  阅读(65)  评论(0编辑  收藏  举报