注意
trie树的空间复杂度较高
因为trie树是一棵二叉树由于每一层都有2i个数,因此一共需要2n+1个空间来存储。另外本题需要id数组来存储下标。也就是最后一层所有存在的数都对应着原数组的下标。因此id数组也需要和trie一样的存储空间。
由于异或两次相当于没有异或,因此可以用前缀和来存储。通过端点的异或就可以计算出区间的异或值。
问题描述
农夫约翰在给他的奶牛们喂食时遇到了一个问题。
他共有 N 头奶牛,编号 1∼N。
每次喂食前,这 N 头奶牛会按照 1∼N 的顺序站成一排。
此外,每头奶牛都被分配了一个可能不唯一的整数。
那么所有被分配的整数就形成了一个长度为 N 的整数序列。
请你在该整数序列中找出一个连续的非空子序列,使得子序列中元素的异或和能够最大。
如果存在多个这样的序列,那么选择序列末端整数对应的奶牛编号更小的那个序列。
如果仍然存在多个可选的序列,那么选择长度最短的那个序列。
输入格式
第一行包含整数 N。
第 2∼N+1 行,每行包含一个整数,其中第 i 行的整数表示编号为 i−1 的牛被分配的整数值。
输出格式
输出三个整数,分别表示最大的异或和,所选序列首端整数对应的奶牛编号,所选序列末端整数对应的奶牛编号。
数据范围
1≤N≤105,
分配给奶牛的整数的范围是 [0,221−1]。
输入样例:
5
1
0
5
4
2
输出样例:
6 4 5
代码
#include<iostream>
using namespace std;
const int N = 100010, M = N * 21;
int idx = 0, trie[M][2];
int id[M];
int s[N];
void insert(int x, int k){
int q = 0;
for(int i = 20; i >= 0; --i){
int u = x >> i & 1;
if(!trie[q][u]) trie[q][u] = ++idx;
q = trie[q][u];
}
id[q] = k;
}
int query(int x){
int q = 0;
for(int i = 20; i >= 0; --i){
int u = x >> i & 1;
if(trie[q][!u]){
q = trie[q][!u];
}else{
q = trie[q][u];
}
}
return id[q];
}
int main(){
int n;
scanf("%d", &n);
for(int i = 1; i <= n; ++i){
scanf("%d", &s[i]);
s[i] ^= s[i - 1];
}
int cnt = -1, a, b;
insert(s[0], 0);
for(int i = 1; i <= n; ++i){
int k = query(s[i]);
int t = s[k] ^ s[i];
if(t > cnt){
a = k + 1;
b = i;
cnt = t;
}
insert(s[i], i);
}
cout << cnt << " " << a << " " << b << endl;
return 0;
}