3579 数字移动（环图、找环、并查集）

1. 问题描述：

将 1∼n 按顺序排成一排，构成一个数列。数字 i 刚好位于位置 i。再给定一个长度为 n 的位置序列 p1，p2，…，pn，它是 1∼n 的一种排列。接下来，我们会重复不断地对数列进行如下操作：重新排列数列中每个数的位置，将位于位置 i 的数移动至 位置 pi。（如果 i = pi 则该数仍移动至位置 i）。每次操作开始时，所有数的移动同时进行，操作结束后，数列将变为一个新的 1∼n 的排列。例如，当 n = 6 并且 p = [4，6，1，3，5，2] 时，第一次操作后，数字 1 将移动至位置 4，数字 2 将移动至位置 6，以此类推；第二次操作后，数字 1 将移动至位置 3，数字 2 将移动至位置 2，以此类推。你的任务是确定从 1 到 n 的每个数字 i，经过多少次操作后，第一次重新回到位置 i。例如，考虑 p = [5，1，2，4，3]，数字 1 的移动轨迹如下：
第一次操作后，到达位置 5。
第二次操作后，到达位置 3。
第三次操作后，到达位置 2。
第四次操作后，回到位置 1。
所以，经过四次操作后，数字 1 第一次回到位置 1。值得一提的是，数字 4 经过一次操作后就回到了位置 4.

输入格式

第一行包含整数 T，表示共有 T 组测试数据。每组数据第一行包含整数 n。第二行包含 n 个整数 p1，…，pn。

输出格式

每组数据输出一行结果，包含 n 个整数，其中第 i 个整数表示数字 i 第一次回到位置 i 所经过的操作次数。整数之间用单个空格隔开。

数据范围

对于 30% 的数据，1 ≤ T ≤ 10，1 ≤ n ≤ 10。
对于 100% 的数据，1 ≤ T ≤ 1000，1 ≤ n ≤ 2 × 10 ^ 5，1 ≤ pi ≤ n。
保证 p1∼pn 是 1∼n 的一种排列。
保证 ∑n ≤ 2 × 10 ^ 5（一个输入中的 T 个 n 相加之和不超过 2 × 10 ^ 5）。

输入样例：

6
5
1 2 3 4 5
3
2 3 1
6
4 6 2 1 5 3
1
1
4
3 4 1 2
5
5 1 2 4 3

输出样例：

1 1 1 1 1 
3 3 3 
2 3 3 2 1 3 
1 
2 2 2 2 
4 4 4 1 4 
来源：https://www.acwing.com/problem/content/3582/

2. 思路分析：

分析题目可以知道本质上是环图问题（转化为图论中的问题：由若干个环构成的图），根据下面的例子会比较好理解，我们需要在图中找出若干个环，可以发现某个数字i经过若干次操作第一次回到位置i对应的操作次数为当前数字所在环的长度，而且环中每一个数字对应的操作次数都为环的长度，每一个环中的数字对应的出度和入度都为1，而每一个环相当于是一个连通块，所以我们可以使用并查集将在同一个环中的所有节点合并到同一个集合，由于需要记录连通块的长度，我们可以声明一个数组s，s表示节点i所在的连通块的长度，一开始的时候所有节点的长度都为1，依次枚举当前的位置i与p[i]，如果两个节点不在同一个集合说明需要将这两个点合并到同一个集合中，我们可以规定其中一个点为父节点x，将另外一个点y合并到当前这个点x所在的集合中，并且节点x所在集合中点的数目需要加上节点y所在集合钟点的数目，最后输出每一个点所在连通块的点的数目即可。

3. 代码如下：

from typing import List


class Solution:
    # 并查集查找节点x的父节点并且进行路径压缩
    def find(self, x: int, fa: List[int]):
        if x != fa[x]: fa[x] = self.find(fa[x], fa)
        return fa[x]

    # 本质上是环图问题, 可以使用并查集来解决
    def process(self):
        T = int(input())
        while T > 0:
            n = int(input())
            p = list(map(int, input().split()))
            T -= 1
            # s[i]表示节点i所在连通块中点的数目
            fa, s = [i for i in range(n + 10)], [1] * (n + 10)
            for i in range(1, n + 1):
                j = p[i - 1]
                pa, pb = self.find(i, fa), self.find(j, fa)
                if pa != pb:
                    # 当前pa是父节点所以需要将另外一个节点的集合数目累加到当前的父节点中
                    s[pa] += s[pb]
                    fa[pb] = pa
            for i in range(1, n + 1):
                print(s[self.find(i, fa)], end=" ")
            print()


if __name__ == '__main__':
    Solution().process()