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

题目连接

题面：

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#define ll long long
using namespace std;
const int maxn=1001000;
const int mod=998244353;
const int inv2=499122177;
int a[maxn],b[maxn],x1[maxn],x2[maxn];
int n;

void fwt_or_mod(int *x,int len,int opt)
{
    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++)
            {
                if(opt==1) x[i+j+k]=(x[j+k]+x[i+j+k])%mod;
                else x[i+j+k]=(x[i+j+k]+mod-x[j+k])%mod;
            }
        }
    }
}

void fwt_and_mod(int *x,int len,int opt)
{
    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++)
            {
                if(opt==1) x[j+k]=(x[j+k]+x[i+j+k])%mod;
                else x[j+k]=(x[j+k]+mod-x[i+j+k])%mod;
            }
        }
    }
}

void fwt_xor_mod(int *x,int len,int opt)
{
    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 xx=x[j+k],yy=x[i+j+k];
                x[j+k]=(xx+yy)%mod;
                x[i+j+k]=(xx+mod-yy)%mod;
                if(opt==-1)
                    x[j+k]=1ll*x[j+k]*inv2%mod,x[i+j+k]=1ll*x[i+j+k]*inv2%mod;
            }
        }
    }
}

void fi_or(int *a,int *b)
{
    for(int i=0;i<(1<<n);i++)
        x1[i]=a[i],x2[i]=b[i];
    fwt_or_mod(x1,1<<n,1);
    fwt_or_mod(x2,1<<n,1);

    for(int i=0;i<(1<<n);i++)
       x1[i]=1ll*x1[i]*x2[i]%mod;
    fwt_or_mod(x1,1<<n,-1);

    for(int i=0;i<(1<<n);i++)
        printf("%d ",x1[i]);
    putchar('\n');
}

void fi_and(int *a,int *b)
{
    for(int i=0;i<(1<<n);i++)
        x1[i]=a[i],x2[i]=b[i];
    fwt_and_mod(x1,1<<n,1);
    fwt_and_mod(x2,1<<n,1);

    for(int i=0;i<(1<<n);i++)
       x1[i]=1ll*x1[i]*x2[i]%mod;
    fwt_and_mod(x1,1<<n,-1);

    for(int i=0;i<(1<<n);i++)
        printf("%d ",x1[i]);
    putchar('\n');
}

void fi_xor(int *a,int *b)
{
    for(int i=0;i<(1<<n);i++)
        x1[i]=a[i],x2[i]=b[i];
    fwt_xor_mod(x1,1<<n,1);
    fwt_xor_mod(x2,1<<n,1);

    for(int i=0;i<(1<<n);i++)
       x1[i]=1ll*x1[i]*x2[i]%mod;
    fwt_xor_mod(x1,1<<n,-1);

    for(int i=0;i<(1<<n);i++)
        printf("%d ",x1[i]);
    putchar('\n');
}

int main(void)
{
    scanf("%d",&n);
    for(int i=0;i<(1<<n);i++)
        scanf("%d",&a[i]);
    for(int i=0;i<(1<<n);i++)
        scanf("%d",&b[i]);
    fi_or(a,b);
    fi_and(a,b);
    fi_xor(a,b);
    return 0;
}