【ACWing】3485. 最大异或和

题目地址：

https://www.acwing.com/problem/content/description/3488/

给定一个非负整数数列$a$，初始长度为$N$。请在所有长度不超过$M$的连续子数组中，找出子数组异或和的最大值。子数组的异或和即为子数组中所有元素按位异或得到的结果。注意：子数组可以为空。

输入格式：
第一行包含两个整数$N,M$。
第二行包含$N$个整数，其中第$i$个为$a_i$。

输出格式：
输出可以得到的子数组异或和的最大值。

数据范围：
对于$20$的数据，$1\le M\le N\le 100$
对于$50$的数据，$1\le M\le N\le 1000$
对于$100$的数据，$1\le M\le N\le 10^5,0\le a_i\le 2^{31}-1$

思路是$0-1$Trie。将每个数看成二进制的$31$位整数（因为$a_i\ge 0$所以最高位的符号位为$0$）。先求出前缀异或数组$b$，即$b[i]=a_0\wedge a_1\wedge ...\wedge a_i$，这样要求以$b[k]$为右端点的，长度不超过$M$的最大的区间异或和，其实就相当于问$b[k-m:k-1]$中哪个数与$b[k]$异或值最大，这可以用Trie来做。同时我们需要将出窗口的值在Trie里删掉，这可以在每个节点维护一个计数，这个计数是经过该节点的元素个数。删除的时候更新计数即可。代码如下：

#include <iostream>
using namespace std;

const int N = 30e5 + 10;
int n, m;
int a[N];
int son[N][2], cnt[N];
int idx;

void insert(int x, int add) {
    int p = 0;
    for (int i = 30; i >= 0; i--) {
        int c = x >> i & 1;
        if (!son[p][c]) son[p][c] = ++idx;
        p = son[p][c];
        cnt[p] += add;
    }
}

int query(int x) {
    int p = 0, res = 0;
    for (int i = 30; i >= 0; i--) {
        int c = x >> i & 1;
        // 看看有没有!c这条路可以走，如果有则走
        if (cnt[son[p][!c]]) {
            p = son[p][!c];
            res |= 1 << i;
        } else p = son[p][c];
    }

    return res;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        a[i] ^= a[i - 1];
    }

    int res = 0;
    insert(0, 1);
    for (int i = 1; i <= n; i++) {
        if (i - m - 1 >= 0) insert(a[i - m - 1], -1);
        res = max(res, query(a[i]));
        insert(a[i], 1);
    }

    printf("%d\n", res);

    return 0;
}

时空复杂度$O(N)$。