2020牛客多校第二场E题 Exclusive OR FWT

Exclusive OR

题意

$n$个数$a_1,a_2,...,a_n$, 求从中取出$i$个数的最大异或和，输出$i=1,2,...,n$的答案

题解

FWT：$C_x=\sum _{i\oplus j=x}{A}_i{B}_j$
这是一个卷积的形式，只不过平常的卷积运算符是$+$
我们用$f_1[x]$记录$a_1,a_2,...,a_n$中$x$是否出现
那么$f_2[x]=\sum _{i\oplus j=x}{f}_1[i]f_1[j]$代表的含义就是

选取两个数$i,j$的异或值为$x$的种数

同理，$f_k[x]=\sum _{i\oplus j=x}{f}_{k-1}[i]f_1[j]$代表的含义为

在选取了$k-1$个数能得到的异或和数中选取一个$i$，在初始数字中选取一个$j$, 使得$i\oplus j=x$的种数

所以这就是一个FWT的题
但是我们不需要知道取得$x$的方案数有多少种，我们只关心能不能取到，所以每次卷积完之后，都将能取到的$x$设为$1$

另外当$i>19$的情况，题目说的比较清楚了

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX = 2e5 + 10;

int N;
int a[1 << 18], b[1 << 18], n;
int ans[MAX];

void get() { for (int i = 0; i < n; i++) a[i] *= b[i]; }

void XOR(int *f, int x = 1) {
    for (int o = 2, k = 1; o <= n; o <<= 1, k <<= 1)
        for (int i = 0; i < n; i += o)
            for (int j = 0; j < k; j++) {
                int a0 = f[i + j], a1 = f[i + j + k];
                f[i + j] = a0 + a1;
                f[i + j + k] = a0 - a1;
                if (x == -1) {
                    f[i + j] /= 2;
                    f[i + j + k] /= 2;
                }
            }
}


int main() {

    scanf("%d", &N); n = 1 << 18;
    for (int i = 1; i <= N; i++) {
        int x; scanf("%d", &x);
        a[x] = b[x] = 1;
        ans[1] = max(ans[1], x);
    }
    XOR(b,1);
    for (int i = 2; i <= 20; i++) {
        XOR(a, 1); get(); XOR(a, -1);
        for (int j = 0; j < n; j++)
            if (a[j]) {
                a[j] = 1;
                ans[i] = max(ans[i], j);
            }
    }
    for (int i = 20; i <= N; i++)
        ans[i] = ans[i - 2];
    for (int i = 1; i <= N; i++)
        printf("%d ", ans[i]);

    return 0;
}