AtCoder AGC031D A Sequence of Permutations (群论、置换快速幂)

题目链接

https://atcoder.jp/contests/agc031/tasks/agc031_d

题解

这居然真的是个找规律神题。。。

首先要明白置换的一些基本定义，置换$p$和$q$的复合$a$定义为$a_i=p_$, 记作$a=pq$. 有定理$(pq){-1}p^{-1}$.
显然题目里定义的$f(p,q)=qp^{-1}$.
然后打表打出前几项:
\(a_1=p\)
\(a_2=q\)
\(a_3=qp^{-1}\)
\(a_4=qp^{-1}q^{-1}\)
\(a_5=qp^{-1}q^{-1}pq^{-1}\)
\(a_6=qp^{-1}q^{-1}p^2q^{-1}\)
\(a_7=qp^{-1}q^{-1}pqpq^{-1}\)
\(a_8=qp^{-1}q^{-1}pqp^{-1}qpq^{-1}\)
好像……规律并不明显啊……
好吧，结论是$a_n=ga_g^{-1}$, 其中$g=qp^{-1}q^{-1}p$. (这是怎么看出来的……)
知道了结论，我们还是比较容易归纳证明的: 显然$gg^{-1}=e$ ($e$为单位元，\(e_i=i\)), 于是$a_n=(ga_g^{-1})(ga_g^{-1}){-1}g^{-1}=ga_g^{-1}$.
于是设$m'=\lfloor \frac{6}\rfloor, n'=m-6m'$, \(a_n=g^{m'}a_{n'}g^{-m'}\), 直接快速幂计算即可，时间复杂度$O(n\log m)$或$O(n)$.

代码

#include<cstdio>

#include<cstdlib>

#include<cassert>

#include<iostream>

using namespace std;

inline int read()

{

	int x=0; bool f=1; char c=getchar();

	for(;!isdigit(c);c=getchar()) if(c=='-') f=0;

	for(; isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+(c^'0');

	if(f) return x;

	return -x;

}

const int N = 1e5;

const int lgM = 30;

int p[N+3],q[N+3],pp[N+3],qq[N+3];

int g[N+3],f[N+3],ff[N+3];

int tmp[N+3];

int aux[N+3];

int ans[N+3];

int a[7][N+3];

int n,m;

int main()

{

	scanf("%d%d",&n,&m);

	for(int i=1; i<=n; i++) scanf("%d",&p[i]),pp[p[i]] = i;

	for(int i=1; i<=n; i++) scanf("%d",&q[i]),qq[q[i]] = i;

	for(int i=1; i<=n; i++) a[1][i] = p[i],a[2][i] = q[i];

	for(int k=3; k<=6; k++)

	{

		for(int i=1; i<=n; i++) a[k][a[k-2][i]] = a[k-1][i];

	}

	for(int i=1; i<=n; i++) g[i] = q[pp[qq[p[i]]]],f[i] = i,tmp[i] = g[i];

	int nn = m%6==0?6:m%6; m = (m-1)/6;

	for(int i=0; m; i++)

	{

		if(m&(1<<i))

		{

			m-=(1<<i);

			for(int j=1; j<=n; j++) aux[j] = f[tmp[j]];

			for(int j=1; j<=n; j++) f[j] = aux[j];

		}

		for(int j=1; j<=n; j++) aux[j] = tmp[tmp[j]];

		for(int j=1; j<=n; j++) tmp[j] = aux[j];

	}

	for(int i=1; i<=n; i++) ff[f[i]] = i;

	for(int i=1; i<=n; i++) ans[i] = f[a[nn][ff[i]]];

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

	return 0;

}