快速沃尔什变换 [FWT变换] 模板

原文链接
FWT变换是可以求: 给定两个数组 A 和 B (长度为2的整数幂):
Ck=ij=kAi·Bi C k = ∑ i ⊕ j = k A i · B i
其中 可以是与,或,异或. 把 A 和 B 分别正变换,在一个好的复杂度得到 A 与 B 按位卷积的的正变换,最后再逆变换回来就是答案。
模板

//注意如果取模的话,可能会有负值
ll add(ll x, ll y){return (x += y) >= MOD ? x - MOD : x;}
ll sub(ll x, ll y){return (x -= y) < 0 ? x + MOD : x;}

void FWT_or(ll a[], int n, int on) {
    for(int i = 1; i < n; i <<= 1) {
        for(int j = 0; j < n; j += (i<<1)) {
            for(int k = 0; k < i; ++k) {
                int u = a[j + k], t = a[j + k + i];
                a[j + k] = u;
                if(on == 1) a[j + k + i] = add(t, u);
                else a[j + k + i] = sub(t, u);
            }
        }
    }
}
void FWT_and(int a[], int n, int on) {
    for(int i = 1; i < n; i <<= 1) {
        for(int j = 0;j < n; j += (i << 1)) {
            for(int k = 0; k < i; ++k) {
                int u = a[j + k], t = a[j + k + i];
                if(on == 1) a[j + k] = add(u, t);
                else a[j + k] = sub(u, t);
                a[j + k + i] = t;
            }
        }
    }
}
void FWT_xor(int a[],int n,int on) {
    for(int i = 1; i < n; i <<= 1) {
        for(int j = 0; j < n; j += (i << 1)) {
            for(int k = 0; k < i; ++k) {
                int u = a[j + k], t = a[j + k + i];
                a[j + k]=add(u, t); a[j + k + i] = sub(u, t);
                if(on == -1) {
                    a[j + k] = (ll)a[j+k] * inv2 % MOD;
                    a[j + k + i] = (ll)a[j + k + i] * inv2 % MOD;

                }
            }
        }
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值