题目描述
输入描述:
输出描述:
示例1
输入
3 998244353
2 3 1
输出
3 1 2
题目大意
给定1~n的大小为n的数列以及置换的次数,再有一个结果数列,求一次置换后的数列是什么
分析1
题意直观解释,以样例为例。
看官们可以先看下这篇论文里面有第二种思路,这里就不先赘述了。接下来先看官方的题解方式。
首先看到这题,肯定是先手动去模拟下。
多试几组数据,可以发现,多次转置后,可能会出现分成不同的团,这些团的内部不断的循环转置,易得,当团转置其长度的倍数次时,会变回原数列,因此k可以直接mod掉(毕竟1e9不是开玩笑的)。我们可以设每团原数列为T,转置的次数即为T的指数,因此,我们知道了T ^k,要求T ^1。看到1,就想到了乘法逆元。
可以设k关于数列长度p的逆元为inv,
则k *inv=1,。
所以T ^1=(T ^k) ^inv=T ^k *inv。
于是重点即为求k的逆元,首先考虑费马小定理,但是用它求的话p必须是素数,然而p是团的长度,是不一定为素数的。因此只能用EXGCD求解同余方程求逆元,EXGCD的要求是k和p互质,而k是比n大的素数,必定与n互质,也一定与p互质。
随后,求出了k的逆元inv,问题就是怎么把T ^k再转置inv次。
其实很简单,只要用vector存入团中的点,然后一一对应过去到答案数组即可。具体见代码。
代码
#include<bits/stdc++.h>
#define ll long long
#define inf 1<<30
using namespace std;
const int MAXN=1e5+100;
vector<ll> vec