[学习笔记]快速沃尔什变换 (FWT)

FWT的简介

一般$\text{FWT}$用来解决一下问题：

• $C_k={\sum }_{i|j=k}{A}_i{B}_j$
• $C_k={\sum }_{i\&j=k}{A}_i{B}_j$
• $C_k={\sum }_{i\oplus j=k}{A}_i{B}_j$

实现的大概思路就是就是先把他们转化成$\text{fwt(A)}$（类似$\text{FFT}$的点值表达），然后对应为相乘，最后在还原为多项式（整个过程很类似与快速傅里叶变换）

主要的就是我们设计的正变换要满足可以直接相乘得到 $C$ 的正变换，比如 $or$ 卷积的证明：
$$C_i^{\prime }=\sum _{j\in i}{C}_j=\sum _{j\in i}\sum _{a|b=j}{A}_a{B}_b$$$$=\sum _{a\in i}{A}_a\sum _{b\in i}{B}_b=A_a^{\prime }{B}_b^{\prime }$$

or 卷积

现在要做到这个的快速卷积：$C_k={\sum }_{i|j=k}{A}_i{B}_j$

定义$A|B$为多项式的$\text{or}$卷积，显然 $A|B=B|A$（交换律），$(A+B)|C=A|C+B|C$（结合律）

定义$fwt(A)[k]={\sum }_{i|k}{A}_i$，因为我们要让$fwt(C)=fwt(A)\times fwt(B)$，这个基于的原理就是若$i|k=k$，$j|k=k$，就能推出$(i|j)|k=k$，那么原来是$k$子集的或之后还是$k$子集。

现在我们来研究如何计算$fwt(A)$，类似$\text{FFT}$地用分治的方法，设$A_0$为当前位为$0$的项，$A_1$就是剩下部分：
$$fwt(A)=\{\begin{array}{ll}(fwt(A_0),fwt(A_0+A_1))& n>0\\ A& n=0\end{array}$$根据$fwt(A)$的定义可以知道：$fwt(A+B)=fwt(A)+fwt(B)$，根据定义也可以知道如果最高位为$1$，那么就把$fwt(A_0+A_1)$算作$A$的后半部分就可以了，为$0$的话子集就是$0$

现在我们来证明一下$fwt(A|B)=fwt(A)\times fwt(B)$：
$$\begin{gathered} fwt(A|B)=fwt((A|B{)}_0,(A|B{)}_1) \\ =fwt(A_0|B_0,A_0|B_1+A_1|B_0+A_1|B_1) \\ =(fwt(A_0|B_0),fwt(A_0|B_0+A_1|B_0+A_0|B_1+A_1{B}_1)) \\ =(fwt(A_0)\times fwt(B_0),fwt(A_0+A_1)\times fwt(B_0+B_1)) \\ =(fwt(A_0),fwt(A_0+A_1))\times (fwt(B_0),fwt(B_0,B_1)) \\ =fwt(A)\times fwt(B) \end{gathered}$$这里用到了数学归纳法，首先$n=0$的情况肯定成立，然后我们假设较小的规模成立，以此推导更大的规模（这里是$\frac{1}{2}$）

最后返回来的$dfwt$就是根据上面的$fwt$设计的，变换如下：
$$dfwt(A)=\{\begin{array}{ll}dfwt(A_0),dfwt(A_1-A_0)& n>1\\ A& n=0\end{array}$$

and 卷积

这个和上面的卷积极其类似，直接给出结论：
$$fwt(A)=\{\begin{array}{ll}(fwt(A_0+A_1),fwt(A_0))& n>0\\ A& n=0\end{array}$$逆变换如下：
$$dfwt(A)=\{\begin{array}{ll}dfwt(A_0-A_1),dfwt(A_1)& n>0\\ A& n=0\end{array}$$

xor 卷积

这就是重头戏了，我们要解决：$C_k={\sum }_{i\oplus j=k}{A}_i{B}_j$

先定义$fwt(A)[x]={\sum }_{2|d(x\cap i)}{A}_i-{\sum }_{2|[d(x\cap i)-1]}{A}_i$，$d$是二进制$1$的个数，这样定义是基于一个结论：
$$d(x\cap (i\oplus j))=d(x\cap i)+d(x\cap j)-2d(x\cap i\cap j)$$考虑每一位的合法性，就可以推知所有情况，这个自己枚举一下可能的组合就行了，然后：
$$fwt(A)[x]\oplus fwt(B)[x]=\sum _{2|d(x\cap i)}\sum _{2|d(x\cap j)}{A}_i{B}_j-\sum _{2|[d(x\cap i)-1]}\sum _{2|d(x\cap j)}{A}_i{B}_j......$$你可以发现如果$d(x\cap i)$和$d(x\cap j)$奇偶性相同的话那么前面的符号是正，否则前面的符号是负，我们观察上面的结论，发现$d(x\cap (i\oplus j))$和$d(x\cap i)+d(x\cap j)$的奇偶性相同，恰好就对应了。

现在来考虑正变换，基本思路还是考虑最高位，左半边的话都选自己和选自己再选对半边都可以，而且不需要变号，因为不会产生新的$1$位，两个都选右半边的话就需要变号，那么就这样算：
$$fwt(A)=\{\begin{array}{ll}(fwt(A_0+A_1),fwt(A_0-A_1))& n>0\\ A& n=0\end{array}$$然后逆变化就是：$$ifwt(A)=\{\begin{array}{ll}(\frac{ifwt(A_0+A_1)}{2},\frac{ifwt(A_0-A_1)}{2})& n>0\\ A& n=0\end{array}$$
然后贴一个板题的代码：

#include <cstdio>
const int M = 200005;
const int MOD = 998244353;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,a[M],b[M],A[M],B[M],inv2=(MOD+1)/2;
void fwt_or(int *a,int op)
{
	for(int i=1;i<n;i<<=1)
		for(int p=i<<1,j=0;j<n;j+=p)
			for(int k=0;k<i;k++)
			{
				if(op==1) a[i+j+k]=(a[i+j+k]+a[j+k])%MOD;
				else a[i+j+k]=(a[i+j+k]-a[j+k]+MOD)%MOD;
			}
}
void fwt_and(int *a,int op)
{
	for(int i=1;i<n;i<<=1)
		for(int p=i<<1,j=0;j<n;j+=p)
			for(int k=0;k<i;k++)
			{
				if(op==1) a[j+k]=(a[j+k]+a[i+j+k])%MOD;
				else a[j+k]=(a[j+k]-a[i+j+k]+MOD)%MOD;
			}
}
void fwt_xor(int *a,int op)
{
	for(int i=1;i<n;i<<=1)
		for(int p=i<<1,j=0;j<n;j+=p)
			for(int k=0;k<i;k++)
			{
				int x=a[j+k],y=a[i+j+k];
				a[j+k]=(x+y)%MOD;
				a[i+j+k]=(x+MOD-y)%MOD;
				if(op==-1)
				{
					a[j+k]=1ll*a[j+k]*inv2%MOD;
					a[i+j+k]=1ll*a[i+j+k]*inv2%MOD;
				}
			}
}
void init()
{
	for(int i=0;i<n;i++)
		A[i]=a[i],B[i]=b[i];
}
signed main()
{
	n=1<<read();
	for(int i=0;i<n;i++) a[i]=read();
	for(int i=0;i<n;i++) b[i]=read();
	init();
	fwt_or(A,1);fwt_or(B,1);
	for(int i=0;i<n;i++) A[i]=1ll*A[i]*B[i]%MOD;
	fwt_or(A,-1);
	for(int i=0;i<n;i++) printf("%d ",A[i]);
	puts("");
	init();
	fwt_and(A,1);fwt_and(B,1);
	for(int i=0;i<n;i++) A[i]=1ll*A[i]*B[i]%MOD;
	fwt_and(A,-1);
	for(int i=0;i<n;i++) printf("%d ",A[i]);
	puts("");
	init();
	fwt_xor(A,1);fwt_xor(B,1);
	for(int i=0;i<n;i++) A[i]=1ll*A[i]*B[i]%MOD;
	fwt_xor(A,-1);
	for(int i=0;i<n;i++) printf("%d ",A[i]);
	puts("");
}