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

题目链接

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

题解

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

首先要明白置换的一些基本定义，置换$p$和$q$的复合$a$定义为$a_i=p_$, 记作$a=pq$. 有定理$(pq)^{-1}=q{-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}=ga_a_{-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;
}