FWT模板代码

2 篇文章 0 订阅

opt是1表示正变换,-1表示逆变换。原理证明参考https://www.cnblogs.com/cjyyb/p/9065615.html
实际这个板子也是从那里抄的。但是知道原理没啥用,过一段就忘了。
题目链接:https://ac.nowcoder.com/acm/contest/5667/E

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int N;
const int NN=2e5+10;
const int MM=(1<<19)+10;
long long a[MM],b[MM];
void FWT_or(long long *a,int opt)
{
    for(int i=1;i<N;i<<=1)
        for(int p=i<<1,j=0;j<N;j+=p)
            for(int k=0;k<i;++k)
                if(opt==1)a[i+j+k]=(a[j+k]+a[i+j+k]);
                else a[i+j+k]=(a[i+j+k]-a[j+k]);
}
void FWT_and(long long *a,int opt)
{
    for(int i=1;i<N;i<<=1)
        for(int p=i<<1,j=0;j<N;j+=p)
            for(int k=0;k<i;++k)
                if(opt==1)a[j+k]=(a[j+k]+a[i+j+k]);
                else a[j+k]=(a[j+k]-a[i+j+k]);
}
void FWT_xor(long long *a,int opt)
{
    for(int i=1;i<N;i<<=1)
        for(int p=i<<1,j=0;j<N;j+=p)
            for(int k=0;k<i;++k)
            {
                long long X=a[j+k],Y=a[i+j+k];
                a[j+k]=(X+Y);a[i+j+k]=(X-Y);
                if(opt==-1)a[j+k]=1ll*a[j+k]/2,a[i+j+k]=1ll*a[i+j+k]/2;
            }
}
int ans[NN];
int main() {
    int n;scanf("%d",&n);
    int maxn=0;
    for(int i=1;i<=n;i++){
        int x;scanf("%d",&x);
        maxn=max(maxn,x);
        b[x]++;
    }
    for(N=1;N<maxn;N<<=1);N<<=1;
    a[0]=1;
    FWT_xor(b,1);
    for(int i=1;i<=19;i++){
        FWT_xor(a,1);
        for(int j=0;j<N;j++)a[j]=a[j]*b[j];
        FWT_xor(a,-1);
        for(int j=0;j<N;j++){
            if(a[j]>0){a[j]=1;ans[i]=j;}
        }
    }
    for(int i=1;i<=n;i++){
        if(i>19){
            if((i-19)%2){
                printf("%d ",ans[18]);
            }
            else{
                printf("%d ",ans[19]);
            }
        }
        else{
            printf("%d ",ans[i]);
        }
    }
    return 0;
}
/*
23
1 2 4 8 16 32 64 128 256 512 1024 2048 4096 8192 16384 32768 65536 131072 0 0 0 0 0
*/
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值