【学习笔鸡】快速沃尔什变换FWT

【学习笔鸡】快速沃尔什变换FWT

OR的FWT

快速解决:

\[C[i]=\sum_{j|k=i} A[j]B[k] \]

FWT使得我们

\[FWT(C)=FWT(A)*FWT(B) \]

其中\(*\)是点积，就是对应位置乘起来。

而对于\(orFWT\)，

\[C'[i]=FWT(C)[i]=\sum_{j\subseteq i}C[j] \]

那么证明一下：

\[\begin{array} &C'[i]&=\sum_{j\subseteq i} C[j] \\ &=\sum_{j\subseteq i}\sum_{p|k=j} A[p]B[k] \\ &=\sum_{p\subseteq i,k\subseteq i} A[p]B[k] \\ &=\sum_{p\subseteq i} A[p]\sum_{k\subseteq i}B[k] \\ &=A'[i]B'[i] \end{array} \]

考虑\(A\)和\(A'\)的关系，其中\(A_0,A_1\)分别代表\(A\)的前\(2^{k-1}\)和后这么多项(下标都从0开始)。他们的差别是\(2^{k-1}\)位上的不同。其他相似。

\[FWT(A)= \begin{cases} FWT(A_0),FWT(A_1+A_0) & k>0 \\ A & k=0 \end{cases} \]

逗号表示依次连接。

复杂度\(T(n)=2T(n/2)+T(n)=O(n\log n)\)，而一般来说\(n=2^m\)那么就是\(O(m2^m)\)

考虑\(IFWT\)

照猫画虎即可

\[IFWT(A')= \begin{cases} IFWT(A'_0),IFWT(A'_1-A'_0) & k>0 \\ A' & k=0 \end{cases} \]

代码

inline void FWT_OR(int*a,const int&tag,const int&len){
	for(int t=1;t<len;t<<=1)
		for(int i=0;i<len;i+=t<<1)
			for(int k=0;k<t;++k)
				a[t+i+k]=(a[t+i+k]+a[t+i]*tag+mod)%mod;
}

AND的FWT

同样地，快速解决

\[C[i]=\sum_{j\& k=i}A[j]B[k] \]

可以构造\(C'[i]=FWT(C)[i]=\sum_\limits{i\subseteq j} C[j]\)，至于为什么构造，这个\(j\&k=i\)可以看做\(i\)是\(j,k\)的子集。

同样有

\[FWT(C)=FWT(A)*FWT(B) \]

证明:

\[\begin{array} &C'[i]&=\sum_{i\subseteq j} C[j] \\ &= \sum_{i\subseteq j} \sum_{k\&p=j}A[k]B[p] \\ &= \sum_{i\subseteq k,i\subseteq p}A[k]B[p] \\ &= \sum_{i\subseteq k}A[k]\sum_{i\subseteq p}B[p] \\ &=A'[i]B'[i] \end{array} \]

同样地

\[FWT(A)= \begin{cases} FWT(A_0+A_1),FWT(A_1)&k>0 \\ A&k=0 \end{cases} \]

同样的\(T(n)=O(m2^m)\)

同样地

\[IFWT(A')= \begin{cases} IFWT(A'_0-A'_1),IFWT(A_1) &k>0 \\ A'&k=0 \end{cases} \]

同样地

inline void FWT_AND(int*a,const int&tag,const int&len){
	for(int t=1;t<len;t<<=1)
		for(int i=0;i<len;i+=t<<1)
			for(int k=0;k<t;++k)
				a[t+i]=(a[t+i]+a[t+i+k]*tag+mod)%mod;
}

XOR的FWT

也是快速解决

\[C[i]=\sum_{j\oplus k=i}A[j]B[k] \]

这里\(FWT(X)\)貌似没有很直观的意义了，推式子的话其实也能理解

\[FWT(C)=FWT(A)*FWT(B) \]

这里记录一个符号\(A\oplus B=C\)

那么

\(C=A\oplus B\)

拆成前后两半

\[C_0=A_0\oplus B_0+A_1\oplus B_1 \\ C_1=A_0\oplus B_1+A_1\oplus B_0 \]

令

\[X_0=(A_0+A_1)\oplus (B_0+B_1) \\ X_1=(A_0-A_1)\oplus (B_0-B_1) \]

然后?

\[C_0={X_0+X_1\over 2} \\ C_1={X_0-X_1 \over 2} \]

​

但是好像学这个无法和各路大佬进行交流，并且我好像并没有学会，那么...

\[FWT(A)=\begin{cases}(FWT(A_0)+FWT(A_1),FWT(A_0)-FWT(A_1)) & n>0\\A & n=0\end{cases} \]

用循环实现的技巧和NTT一致，控制长度，控制第几段，控制段内的循环变量

//@winlere
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;  typedef long long ll;
inline int qr(){
	int ret=0,f=0,c=getchar();
	while(!isdigit(c))f|=c==45,c=getchar();
	while(isdigit(c)) ret=ret*10+c-48,c=getchar();
	return f?-ret:ret;
}
const int maxn=1<<18|1;
const int mod=998244353;

inline void FWT_OR(int*a,const int&len,const int&tag){
	for(int t=1;t<len;t<<=1)
		for(int i=0;i<len;i+=t<<1)
			for(int k=0;k<t;++k)
				a[t+i+k]=(0ll+a[t+i+k]+a[i+k]*tag+mod)%mod;
}

inline void FWT_AND(int*a,const int&len,const int&tag){
	for(int t=1;t<len;t<<=1)
		for(int i=0;i<len;i+=t<<1)
			for(int k=0;k<t;++k)
				a[i+k]=(0ll+a[i+k]+a[t+i+k]*tag+mod)%mod;
}

inline void FWT_XOR(int*a,const int&len,const int&tag){
	int opt=tag==1?1:((mod+1)>>1);
	for(int t=1;t<len;t<<=1)
		for(int i=0;i<len;i+=t<<1)
			for(int k=0;k<t;++k){
				int t0=a[i+k],t1=a[i+k+t];
				if(tag==1) a[i+k]=(t0+t1)%mod,a[i+k+t]=(t0-t1+mod)%mod;
				else a[i+k]=1ll*(t0+t1)%mod*opt%mod,a[i+k+t]=1ll*(t0-t1+mod)%mod*opt%mod;
			}
}

int n,k;
int a[maxn],b[maxn],c[maxn];
int A[maxn],B[maxn];

int main(){
#ifndef ONLINE_JUDGE
	freopen("in.in","r",stdin);
	freopen("out.out","w",stdout);
#endif
	n=qr(); 
	for(int t=0;t<1<<n;++t) a[t]=qr();
	for(int t=0;t<1<<n;++t) b[t]=qr();
	size_t s=(1<<n)*4;
		
	memcpy(A,a,s); memcpy(B,b,s);
	FWT_OR(A,1<<n,1); FWT_OR(B,1<<n,1);
	for(int t=0;t<1<<n;++t) c[t]=1ll*A[t]*B[t]%mod;
	FWT_OR(c,1<<n,-1);
	for(int t=0;t<1<<n;++t) printf("%d ",c[t]);
	putchar('\n');
	
	memcpy(A,a,s); memcpy(B,b,s);
	FWT_AND(A,1<<n,1); FWT_AND(B,1<<n,1);
	for(int t=0;t<1<<n;++t) c[t]=1ll*A[t]*B[t]%mod;
	FWT_AND(c,1<<n,-1);
	for(int t=0;t<1<<n;++t) printf("%d ",c[t]);
	putchar('\n');
	
	memcpy(A,a,s); memcpy(B,b,s);
	FWT_XOR(A,1<<n,1); FWT_XOR(B,1<<n,1);
	for(int t=0;t<1<<n;++t) c[t]=1ll*A[t]*B[t]%mod;
	FWT_XOR(c,1<<n,-1);
	for(int t=0;t<1<<n;++t) printf("%d ",c[t]);
	putchar('\n');
		
	return 0;
}