[ZROI2021]序列变换

题目

题目描述
定义一个从序列到序列的映射 $f(a)=b$ 满足
$$b_i=\underset{j=0}{\overset{k-1}{⨁}}{a}_{(i+j)\mathrm{mod}n}$$

其中 $a,b$ 都是下标在 $0$ 到 $n-1$ 之间的非负整数序列。

现在你需要求出 $f^T(a)=f^{T-1}[f(a)](T>1)$，规定 $f^1(a)=f(a)$ 。即对 $a$ 进行 $T$ 次 $f$ 变换后的结果。

数据范围与提示
$n⩽5\times 10^5,T⩽10^{18}$ 。

思路

直接快速幂不可以过。但是很特殊的是，我们快速幂只是在求
$${\left(1+x+x^2+\cdots +x^{k-1}\right)}^T\mathrm{mod}2$$

显然这个式子等于
$${\left(\frac{1-x^k}{1-x}\right)}^T=\left[\sum _{i=0}^T\left(\genfrac{}{}{0ex}{}{T}{i}\right)(-1{)}^i{x}^{ik}\right]\left[\sum _{i=0}^{+\infty }\left(\genfrac{}{}{0ex}{}{i+T-1}{T-1}\right)x^i\right]$$

即分子分母都展开的结果。这个式子感觉不好整，因为组合数太复杂了……

不过，既然是模 $2$ 意义下，很容易让人想起卢卡斯定理，
$$\left(\genfrac{}{}{0ex}{}{n}{m}\right)\mathrm{mod}2=[m\subseteq n]$$

这里用集合符号表示了二进制位的情况。即 $m$ 与 $n$ 的按位与结果是 $m$ 。

不妨设 $T=2^t(t⩾1)$，那么左边那个括号只有两项，右边那个括号只有 $2^t\mid i$ 的项。于是
$$\begin{gathered} \equiv (1+x^{2^{t}k})\left(\sum _{i=0}^{+\infty }{x}^{2^{t}i}\right) \\ \equiv 1+x^{2^t}+x^{2^t\times 2}+\cdots +x^{2^t(k-1)}(\mathrm{mod}2) \end{gathered}$$

这个东西就很有规律了——以 $2^t$ 为步长，每个点变为前面 $k$ 步的值的异或值。作个 $2^t$ 步长的前缀和，进行一次这种变换就是 $\mathcal{O}(n)$ 的了。

问题只是 $T$ 不是二的幂；二进制拆分就完了。复杂度 $\mathcal{O}(n\log T)$ 。

代码

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(int x){
	if(x > 9) writeint(x/10);
	putchar((x-x/10*10)^48);
}

const int MaxN = 500005;
int a[MaxN], b[MaxN], n, k;
bool vis[MaxN]; // the index
void change(long long T){
	if((T %= n) == 0){
		if(k&1) return ; // make no difference
		memset(a,0,n<<2); return ;
	}
	memset(vis,0,n); // clear
	for(int x=0; !vis[x]; ++x){
		int p = 0, all = 0;
		for(int i=x; !vis[i]; i=(i+T)%n)
			all ^= a[i], ++ p, vis[i] = 1;
		int now = 0, fr = (x+k*T)%n;
		for(int i=x; i!=fr; i=(i+T)%n)
			now ^= a[i]; // k-th number
		for(int i=fr; true; i=(i+T)%n){
			now ^= a[(i-k*T%n+n)%n];
			b[i] = (now ^= a[i]);
			if((k/p)&1) b[i] ^= all;
			if((i+T)%n == fr) break;
		}
	}
	memcpy(a,b,n<<2); // copy back
}

int main(){
	n = readint(), k = readint();
	long long T; scanf("%lld",&T);
	for(int i=0; i<n; ++i)
		a[n-1-i] = readint();
	while(T) change(T&-T), T ^= (T&-T);
	writeint(a[n-1]);
	for(int i=1; i<n; ++i){
		putchar(' ');
		writeint(a[n-1-i]);
	}
	return 0;
}

后记

更强力的关系是 $F(x{)}^p\mathrm{mod}p=F(x^p)(p\in \mathbb{P})$，其中 $\mathbb{P}$ 是质数集。证明方法也是利用卢卡斯定理：当 $F(x)$ 是单项式时，显然成立；当 $F(x)$ 有至少两项时，肯定可以拆成 $G(x)+H(x)$，于是
$$[G(x)+H(x){]}^p=\sum _{i=0}^p(\genfrac{}{}{0ex}{}{p}{i})\cdot G(x{)}^i\cdot H(x{)}^{p-i}$$

只有 i ∈ { 0 , p } i\in\{0,p\} i∈{0,p} 时 $\left(\genfrac{}{}{0ex}{}{p}{i}\right)\mathrm{mod}p\ne 0$，所以其实
$$F(x{)}^p\equiv G(x{)}^p+H(x{)}^p(\mathrm{mod}p)$$

根据归纳法（项数从小到大），我们推出
$$F(x{)}^p\equiv G(x^p)+H(x^p)=F(x^p)(\mathrm{mod}p)$$

如果取 $F(x)=f(x{)}^p\equiv f(x^p)(\mathrm{mod}p)$，你就得到了有趣的结论
$$\begin{gathered} [f(x{)}^p{]}^p\equiv f(x^p{)}^p(\mathrm{mod}p) \\ \Rightarrow f(x{)}^{p^2}\equiv f(x^{p^2})(\mathrm{mod}p) \end{gathered}$$

多迭代几次，你就会得到本题中使用的式子。本题是 $p=2$ 的特殊情况。
$$f(x{)}^{p^t}=f\left(x^{p^t}\right)$$

$\text{2022/8/5 update}$：该式子也可以用二项式系数直接计算得到，但现在看来，似乎还没有这种构造方法简洁 😄