[图论] aw3579. 数字移动(图论+并查集+置换群+dfs+aw周赛001_3)

1. 题目来源

链接：3579. 数字移动

2. 题目解析

置换群问题，隶属于群论、图论知识点，在这可以使用并查集偷鸡。

给出一个置换序列，一次操作就是按照这个序列置换一次。

按题建图，将 i 作为出点，pi 作为入点，连一条边，则整个图会由几个简单环构成。每个点回到原来位置的置换次数就和所在环中的边长度一样，也就和环中的点的个数一样。

可将环看为连通块，并查集维护连通块再维护连通块中点的数量就能偷鸡了。

也可以暴搜求环长，参考抽风的题解。

---

图解如下：

---

时间复杂度：$O(n)$

空间复杂度：$O(n)$

---

#include <bits/stdc++.h>

using namespace std;

const int N = 2e5+5;

int n;
int a[N];
int p[N], s[N];

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    int T;
    scanf("%d", &T);
    while (T -- ) {
        scanf("%d", &n);
        for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]), p[i] = i, s[i] = 1;
        
        for (int i = 1; i <= n; i ++ ) {
            int pa = find(a[i]), pb = find(i);
            if (pa != pb) {
                s[pb] += s[pa];
                p[pa] = pb;
            }
        }
        
        for (int i = 1; i <= n; i ++ ) printf("%d ", s[find(i)]);
        puts("");
    }
    
    return 0;
}

---

暴搜求环长，参考抽风的题解:

#include <cstdio>
#include <cstring>

const int N = 200005;

int n, p[N], res[N];
bool st[N];
int xs[N], sz;

// 返回 x 所在环长
int dfs(int x) {
    if (st[x]) return 0;
    st[x] = true;
    xs[sz++] = x; // 将环上所有点存到一个数组中
    return dfs(p[x]) + 1;
}

int main() {
    int task;
    for (scanf("%d", &task); task--; ) {
        scanf("%d", &n);
        memset(st, false, sizeof st);
        for (int i = 1; i <= n; ++i) scanf("%d", p + i);

        for (int i = 1; i <= n; ++i)
            if (!st[i]) {
                int t = dfs(i);
                for (int k = 0; k < sz; ++k) res[xs[k]] = t;
                sz = 0;
            }

        for (int i = 1; i <= n; ++i) printf("%d ", res[i]);
        puts("");
    }

    return 0;
}