洛谷P4717 【模板】快速沃尔什变换（FWT）

传送门

这玩意儿太骚了……

参考了yyb巨佬的

//minamoto
#include<iostream>
#include<cstdio>
#define ll long long
#define add(x,y) ((x+=y)>=mod?x-=mod:x)
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
inline void print(int x){
    if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]=' ';
}
const int N=(1<<17)+5,mod=998244353,inv=499122177;
int n,len,A[N],B[N],R[N];
void FWT_or(int *A,int type){
    for(int i=1;i<len;i<<=1)
    for(int p=i<<1,j=0;j<len;j+=p)
    for(int k=0;k<i;++k)
    add(A[i+j+k],type==1?A[j+k]:mod-A[j+k]);
}
void FWT_and(int *A,int type){
    for(int i=1;i<len;i<<=1)
    for(int p=i<<1,j=0;j<len;j+=p)
    for(int k=0;k<i;++k)
    add(A[j+k],type==1?A[i+j+k]:mod-A[i+j+k]);
}
void FWT_xor(int *A,int type){
    for(int i=1;i<len;i<<=1)
    for(int p=i<<1,j=0;j<len;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(type==-1) A[j+k]=1ll*A[j+k]*inv%mod,A[i+j+k]=1ll*A[i+j+k]*inv%mod;
    }
}
int main(){
//    freopen("testdata.in","r",stdin);
    n=read(),len=1<<n;
    for(int i=0;i<len;++i) A[i]=read();
    for(int i=0;i<len;++i) B[i]=read();
    
    FWT_or(A,1),FWT_or(B,1);
    for(int i=0;i<len;++i) R[i]=1ll*A[i]*B[i]%mod;
    FWT_or(A,-1),FWT_or(B,-1),FWT_or(R,-1);
    for(int i=0;i<len;++i) print(R[i]);sr[++C]='
';
    
    FWT_and(A,1),FWT_and(B,1);
    for(int i=0;i<len;++i) R[i]=1ll*A[i]*B[i]%mod;
    FWT_and(A,-1),FWT_and(B,-1),FWT_and(R,-1);
    for(int i=0;i<len;++i) print(R[i]);sr[++C]='
';
    
    FWT_xor(A,1),FWT_xor(B,1);
    for(int i=0;i<len;++i) R[i]=1ll*A[i]*B[i]%mod;
    FWT_xor(R,-1);
    for(int i=0;i<len;++i) print(R[i]);sr[++C]='
';
    
    Ot();
    return 0;
}