2020牛客暑期多校训练营（第二场） [Just Shuffle]

2020牛客暑期多校训练营（第二场） Just Shuffle

这个题目和之前写的 随便置换 差不多，现在就贴一下代码吧

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
int a[maxn],f[maxn],cnt[maxn],b[maxn],p[maxn];

int Find(int x){
    return x==f[x]?x:f[x]=Find(f[x]);
}

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

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    init(n);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        if(Find(i)!=Find(a[i])) f[i]=a[i];
    }
    for(int i=1;i<=n;i++) cnt[Find(i)]++;
    for(int i=1;i<=n;i++){
        if(!cnt[i]) continue;
        int l = cnt[i],w=i;
        for(int j=0;j<l;j++){
            b[j*1ll*m%l] = w;
            w = a[w];
        }
        for(int j=0;j<l;j++){
            p[b[j]]=b[(j+1)%l];
        }
    }
    for(int i=1;i<n;i++) printf("%d ",p[i]);
    printf("%d\n",p[n]);
    return 0;
}