FWT模板

最新推荐文章于 2021-03-03 15:41:02 发布

ACTerminate

最新推荐文章于 2021-03-03 15:41:02 发布

阅读量593

点赞数

CC 4.0 BY-SA版权

分类专栏： FWT 文章标签： FWT

本文链接：https://blog.youkuaiyun.com/ACTerminate/article/details/79681749

FWT 专栏收录该内容

2 篇文章

订阅专栏

int rev=mod+1>>1;
void FWT(int a[],int n)    
{    
    for(int d=1;d<n;d<<=1)    
        for(int m=d<<1,i=0;i<n;i+=m)    
            for(int j=0;j<d;j++)    
            {    
                int x=a[i+j],y=a[i+j+d];    
                a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;    
                //xor:a[i+j]=x+y,a[i+j+d]=(x-y+mod)%mod;    
                //and:a[i+j]=x+y;    
                //or:a[i+j+d]=x+y;    
            }    
}    
    
void UFWT(int a[],int n)    
{    
    for(int d=1;d<n;d<<=1)    
        for(int m=d<<1,i=0;i<n;i+=m)    
            for(int j=0;j<d;j++)    
            {    
                int x=a[i+j],y=a[i+j+d];    
                a[i+j]=1LL*(x+y)*rev%mod,a[i+j+d]=(1LL*(x-y)*rev%mod+mod)%mod;    
                //xor:a[i+j]=(x+y)/2,a[i+j+d]=(x-y)/2;    
                //and:a[i+j]=x-y;    
                //or:a[i+j+d]=y-x;    
            }    
}    
void solve(int a[],int b[],int n)    
{  
    FWT(a,n);  
    FWT(b,n);  
    for(int i=0;i<n;i++) a[i]=1LL*a[i]*b[i]%mod;    
    UFWT(a,n);  
}