bzoj 4260: Codechef REBXOR（01字典树）

4260: Codechef REBXOR

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 1281  Solved: 560
[Submit][Status][Discuss]

Description

Input

输入数据的第一行包含一个整数N，表示数组中的元素个数。

第二行包含N个整数A1,A2,…,AN。

Output

输出一行包含给定表达式可能的最大值。

Sample Input

5

1 2 3 1 2

Sample Output

6

设q[i]为前i个数的最大区间异或和，p[i]为后n-i+1个数的最大区间异或和

答案就是max(q[i]+p[i+1])

这样问题就转化了

令F[l, r] = a[l]^a[l+1]^…^a[r]

那么F[l, r] = F[0, r]^F[0, l-1]，也就是两个前缀异或和相异或

用字典树维护一下前缀异或和和后缀异或和就行

#include<stdio.h>#include<stdlib.h>#include<algorithm>using namespace std;#define LL long longtypedef struct Trie{Trie *next[2];}Trie;int a[400005], sum[400005], q[400005], p[400005];void Update(Trie *root, int x){int i, now;Trie *p = root;for(i=30;i>=0;i--){now = 0;if(x&(1<<i))now = 1;if(p->next[now]==NULL){Trie *temp = new Trie;temp->next[0] = temp->next[1] = NULL;p->next[now] = temp;}p = p->next[now];}}int Query(Trie *root, int x){int i, now, ans = 0;Trie *p = root;for(i=30;i>=0;i--){now = 1;if(x&(1<<i))now = 0;if(p->next[now]!=NULL){ans += (1<<i);p = p->next[now];}else if(p->next[now^1]!=NULL)p = p->next[now^1];elsereturn ans;}return ans;}void Delete(Trie *root){int i;for(i=0;i<=1;i++){if(root->next[i]!=NULL)Delete(root->next[i]);}free(root);}int main(void){int n, i;LL ans = 0;Trie *root = new Trie;scanf("%d", &n);for(i=1;i<=n;i++){scanf("%d", &a[i]);sum[i] = sum[i-1]^a[i];}root->next[0] = root->next[1] = NULL;Update(root, 0);for(i=1;i<=n;i++){q[i] = max(Query(root, a[i]), q[i-1]);Update(root, a[i]);}free(root);root = new Trie;root->next[0] = root->next[1] = NULL;Update(root, 0);for(i=n;i>=1;i--){p[i] = max(Query(root, a[i]), p[i]);Update(root, a[i]);}for(i=1;i<=n-1;i++)ans = max(ans, (LL)q[i]+p[i+1]);printf("%lld\n", ans);free(root);return 0;}/*21 2*/