2020牛客暑期多校训练营（第二场）J、Just Shuffle（置换群）

题目链接

题面：

题意：
给定一个长度为n的序列A，和一个大质数k。
求出一个置换p，
使得全排列 1----n，置换k次之后会得到序列A。

官方题解：

群论知识：

题解：
对于一个置换群B，它的阶数为环的长度，B⁰就是单位元。
B¹就是B，我们设给定的A的阶数（环的长度）为 r 。
B置换 k 次之后即B^k=A。
那么在环长为 r 的意义下我们令 B¹=B^(k*p)%r=1=A^p
即若，B^k=A 那么存在---->B¹=A^p，其中（k*p）% r==1

若我们能解出p来，那么A^p即为所求的B。
因为k是一个大质数，所以（k*p）% r==1 一定有解。
只需要求出给定的序列里面的每个置换即可。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<unordered_map>
#include<set>
#define ui unsigned int
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
//#define lc (cnt<<1)
//#define rc (cnt<<1|1)
#define len(x)  (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
using namespace std;

const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1e9+7;
const double eps=1e-8;
const double pi=acos(-1.0);
const int hp=13331;
const int maxn=100100;
const int maxm=100100;
const int up=100000;

//k*p=1 (mod r)
//k*x+r*y=1
//gcd(k,r)=1,所以一定有解

int exgcd(int a,int b,int &x,int &y)
{
    if(b==0)
    {
        x=1,y=0;
        return a;
    }
    int d=exgcd(b,a%b,x,y);
    int z=x;
    x=y;
    y=z-y*(a/b);
    return d;
}

int inv(int b,int m)
{
    int x,y;
    int d=exgcd(b,m,x,y);
    return (x%m+m)%m;
}

int a[maxn],ans[maxn],st[maxn];
bool ha[maxn];

int main(void)
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    {
        if(ha[i]) continue;
        int r=0;
        int x=i;
        while(!ha[x])
        {
            st[r++]=x;
            ha[x]=true;
            x=a[x];
        }
        int p=inv(k,r);
        for(int j=0;j<r;j++)
            ans[st[j]]=st[(j+p)%r];
    }
    for(int i=1;i<=n;i++)
        printf("%d ",ans[i]);
    putchar('\n');
    return 0;
}