题目地址:
https://www.acwing.com/problem/content/description/3488/
给定一个非负整数数列 a a a,初始长度为 N N N。请在所有长度不超过 M M M的连续子数组中,找出子数组异或和的最大值。子数组的异或和即为子数组中所有元素按位异或得到的结果。注意:子数组可以为空。
输入格式:
第一行包含两个整数
N
,
M
N,M
N,M。
第二行包含
N
N
N个整数,其中第
i
i
i个为
a
i
a_i
ai。
输出格式:
输出可以得到的子数组异或和的最大值。
数据范围:
对于
20
20%
20的数据,
1
≤
M
≤
N
≤
100
1≤M≤N≤100
1≤M≤N≤100
对于
50
50%
50的数据,
1
≤
M
≤
N
≤
1000
1≤M≤N≤1000
1≤M≤N≤1000
对于
100
100%
100的数据,
1
≤
M
≤
N
≤
1
0
5
,
0
≤
a
i
≤
2
31
−
1
1≤M≤N≤10^5,0≤a_i≤2^{31}−1
1≤M≤N≤105,0≤ai≤231−1
思路是 0 − 1 0-1 0−1Trie。将每个数看成二进制的 31 31 31位整数(因为 a i ≥ 0 a_i\ge 0 ai≥0所以最高位的符号位为 0 0 0)。先求出前缀异或数组 b b b,即 b [ i ] = a 0 ∧ a 1 ∧ . . . ∧ a i b[i]=a_0\land a_1\land ...\land a_i b[i]=a0∧a1∧...∧ai,这样要求以 b [ k ] b[k] b[k]为右端点的,长度不超过 M M M的最大的区间异或和,其实就相当于问 b [ k − m : k − 1 ] b[k-m:k-1] b[k−m:k−1]中哪个数与 b [ k ] b[k] 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 ) O(N) O(N)。