快速沃尔变换 FWT

P4717 【模板】快速沃尔什变换

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define sc(x) scanf("%lld",&(x));
#define si signed
const int maxn=<<+;
int A[maxn];
int B[maxn];
int C[maxn];
#define mod 998244353
#define inv2 ((mod+1)/2)
int n;
int t;
void FWT_or(int *C,int op)
{
    for(int i=;i<n;i<<=){
        for(int p=i<<,j=;j<n;j+=p){
            for(int k=;k<i;k++){
                if(op==)C[i+j+k]=(C[i+j+k]+C[j+k])%mod;
                if(op==-)C[i+j+k]=(C[i+j+k]-C[j+k]+mod)%mod;
            }
        }
    }
}
void FWT_and(int *C,int op)
{
    for(int i=;i<n;i<<=){
        for(int p=i<<,j=;j<n;j+=p){
            for(int k=;k<i;k++){
                if(op==)C[j+k]=(C[i+j+k]+C[j+k])%mod;
                if(op==-)C[j+k]=(C[j+k]+mod-C[i+j+k])%mod;
            }
        }
    }
}
void FWT_xor(int *C,int op)
{
    for(int i=;i<n;i<<=){
        for(int p=i<<,j=;j<n;j+=p){
            for(int k=;k<i;k++){
                int x=C[j+k],y=C[i+j+k];
                C[j+k]=(x+y)%mod;
                C[j+k+i]=(x-y+mod)%mod;
                if(op==-){
                    C[j+k]=C[j+k]*inv2%mod;
                    C[j+k+i]=C[j+k+i]*inv2%mod;
                }
            }
        }
    }
}
si main()
{

    sc(t);
    n=(<<t);
    for(int i=;i<n;i++){
        sc(A[i]);
    }
    for(int i=;i<n;i++){
        sc(B[i]);
    }
    FWT_or(A,);FWT_or(B,);
    for(int i=;i<n;i++){
        C[i]=(A[i]*B[i])%mod;
    }
    FWT_or(A,-);FWT_or(B,-);
    FWT_or(C,-);
    for(int i=;i<n;i++){
        cout<<C[i]<<' ';
    }
    cout<<'\n';
    FWT_and(A,);FWT_and(B,);
     for(int i=;i<n;i++){
        C[i]=(A[i]*B[i])%mod;
    }
    FWT_and(A,-);FWT_and(B,-);
    FWT_and(C,-);
    for(int i=;i<n;i++){
        cout<<C[i]<<' ';
    }
    cout<<'\n';
    FWT_xor(A,);FWT_xor(B,);
     for(int i=;i<n;i++){
        C[i]=(A[i]*B[i])%mod;
    }
    FWT_xor(A,-);FWT_xor(B,-);
    FWT_xor(C,-);
    for(int i=;i<n;i++){
        cout<<C[i]<<' ';
    }
    cout<<'\n';

}