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

原文链接
FWT变换是可以求: 给定两个数组 A 和 B (长度为2的整数幂)：
$C_k = \sum_{i \oplus j=k}A_i·B_i$
其中$\oplus$可以是与，或，异或. 把 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;

                }
            }
        }
    }
}