FWT 学习笔记
学的时候比较匆忙,于是就学一个 \(\texttt{or,and,xor}\) 卷积跑路。
P4717 【模板】快速莫比乌斯/沃尔什变换 (FMT/FWT)
前置知识:高维前缀和,下面前缀和的操作大多都是用高维前缀和来实现的。
设有两个长度为 \(2^n\) 的序列 \(A,B\),现在我们要对他们进行一下不同类型的卷积。
\(\texttt{or}\) 卷积
\[C_i=\sum_{j~\texttt{or}~k=i}A_j\times B_k \]
考虑 \(n\) 只有 \(1\) 的情况,即 \(A,B\) 的长度都只有 \(2\) 时值怎么样的:
\[C_0=A_0\times B_0\\ C_1=A_0\times B_1+A_1\times B_0+A_1\times B_1\\ C_0+C_1=(A_0+A_1)\times (B_0+B_1) \]
受到面式子的启发,考虑将 \(A,B\) 分别进行一次前缀和,每一个对应为乘起来记为 \(C\),再对 \(C\) 做一遍前缀差即可。
\(\texttt{and}\) 卷积
\[C_i=\sum_{j~\texttt{and}~k=i}A_j\times B_k \]
仍然考虑 \(n\) 只有 \(1\):
\[C_0=A_0\times B_0+A_0\times B_1+A_1\times B_0\\ C_1=A_1\times B_1\\ C_0+C_1=(A_0+A_1)\times (B_0+B_1) \]
将 \(A,B\) 都做一遍后缀和,按位乘起来记为 \(C\),再对 \(C\) 做一遍后缀差即可。
\(\texttt{xor}\) 卷积
\[C_i=\sum_{j~\texttt{xor}~k=i}A_j\times B_k \]
考虑 \(n\) 只有 \(1\):
\[C_0=A_0\times B_0+A_1\times B_1\\ C_1=A_0\times B_1+A_1\times B_0\\ \begin{cases} C_0+C_1=(A_0+A_1)\times (B_0+B_1)\\ C_0-C_1=(A_0-A_1)\times (B_0-B_1) \end{cases} \]
那么根据高维前缀和每一位相减过去即可。
模板题代码
// Author:A weak man named EricQian
#include<bits/stdc++.h>
using namespace std;
#define infll 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define Maxn 200005
#define mod 998244353
#define pb push_back
#define pa pair<int,int>
#define fi first
#define se second
typedef long long ll;
inline int rd()
{
int x=0;
char ch,t=0;
while(!isdigit(ch = getchar())) t|=ch=='-';
while(isdigit(ch)) x=x*10+(ch^48),ch=getchar();
return x=t?-x:x;
}
inline ll maxll(ll x,ll y){ return x>y?x:y; }
inline ll minll(ll x,ll y){ return x<y?x:y; }
inline ll absll(ll x){ return x>0ll?x:-x; }
inline ll gcd(ll x,ll y){ return (y==0)?x:gcd(y,x%y); }
struct FWT
{
int n;
inline int ksm(int x,int y)
{
int ret=1;
for(;y;y>>=1,x=1ll*x*x%mod) if(y&1) ret=1ll*ret*x%mod;
return ret;
}
inline void bitmul(int *a,int *b)
{ for(int i=0;i<n;i++) a[i]=1ll*a[i]*b[i]%mod; }
inline void fwt_or(int *a,int opt)
{
for(int p=2;p<=n;p<<=1) for(int i=0;i<n;i+=p) for(int j=0;j<(p>>1);j++)
(a[i+j+(p>>1)]+=1ll*a[i+j]*opt%mod)%=mod;
}
inline void fwt_and(int *a,int opt)
{
for(int p=2;p<=n;p<<=1) for(int i=0;i<n;i+=p) for(int j=0;j<(p>>1);j++)
(a[i+j]+=1ll*a[i+j+(p>>1)]*opt%mod)%=mod;
}
inline void fwt_xor(int *a,int opt)
{
for(int p=2;p<=n;p<<=1) for(int i=0;i<n;i+=p) for(int j=0;j<(p>>1);j++)
{
int x=a[i+j],y=a[i+j+(p>>1)];
a[i+j]=1ll*(x+y)%mod*opt%mod;
a[i+j+(p>>1)]=1ll*(x-y+mod)%mod*opt%mod;
}
}
}P;
int n,All;
int a[Maxn],b[Maxn],A[Maxn],B[Maxn];
int main()
{
//ios::sync_with_stdio(false); cin.tie(0);
//freopen(".in","r",stdin);
//freopen(".out","w",stdout);
n=rd(),All=1<<n,P.n=All;
for(int i=0;i<All;i++) a[i]=rd();
for(int i=0;i<All;i++) b[i]=rd();
memcpy(A,a,sizeof(a)),memcpy(B,b,sizeof(b));
P.fwt_or(A,1),P.fwt_or(B,1),P.bitmul(A,B),P.fwt_or(A,mod-1);
for(int i=0;i<All;i++) printf("%d ",A[i]);
printf("\n");
memcpy(A,a,sizeof(a)),memcpy(B,b,sizeof(b));
P.fwt_and(A,1),P.fwt_and(B,1),P.bitmul(A,B),P.fwt_and(A,mod-1);
for(int i=0;i<All;i++) printf("%d ",A[i]);
printf("\n");
memcpy(A,a,sizeof(a)),memcpy(B,b,sizeof(b));
P.fwt_xor(A,1),P.fwt_xor(B,1),P.bitmul(A,B),P.fwt_xor(A,P.ksm(2,mod-2));
for(int i=0;i<All;i++) printf("%d ",A[i]);
printf("\n");
//fclose(stdin);
//fclose(stdout);
return 0;
}