AcWing 1414. 牛異或(字首和+字典樹)
阿新 • • 發佈:2021-02-05
技術標籤:字典樹
題目連結:點選這裡
題目大意:
給出一個長度為
n
n
n 的序列
a
1
,
a
2
,
.
.
.
,
a
n
a_1,a_2,...,a_n
a1,a2,...,an ,現需從中選出一段非空連續的序列,使其異或和最大。如果存在多個這樣的序列,選擇序列末端整數對應的下標更小、長度最短的那個序列。
題目分析:
我們先對序列求字首和,此時問題對於字首和序列就轉換成了:從序列中選擇兩個元素,使之異或和最大。此問題可以使用字典樹解決
而題目要求不得隨意輸出一個答案,所以我們需要維護序列的左右端點,我們從左向右掃字首和序列,對於每一個元素只有異或和大於答案才更新,同時在向字典樹插入相同的數時要存下最有一次插入的位置,這樣就可以輸出多個答案中符合題意的那一個解了
具體細節見程式碼:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
#define Inf 0x3f3f3f3f3f3f3f3f
//#define int ll
using namespace std;
int read()
{
int res = 0,flag = 1;
char ch = getchar();
while(ch<'0' || ch>'9')
{
if(ch == '-') flag = -1;
ch = getchar();
}
while(ch>='0' && ch<='9')
{
res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
ch = getchar();
}
return res*flag;
}
const int maxn = 1e5 +5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
int n,sum[maxn],trie[maxn*21][2],id[maxn*21],idx;
void insert(int x,int pos)
{
int p = 0;
for(int i = 20;~i;i--)
{
int u = x>>i&1;
if(!trie[p][u]) trie[p][u] = ++idx;
p = trie[p][u];
}
id[p] = pos;
}
int query(int x)
{
int p = 0;
for(int i = 20;~i;i--)
{
int u = x>>i&1;
if(trie[p][!u]) p = trie[p][!u];
else p = trie[p][u];
}
return id[p];
}
signed main()
{
n = read();
for(int i = 1;i <= n;i++) sum[i] = read(),sum[i] ^= sum[i-1];
int ans = -1,l,r;
for(int i = 1;i <= n;i++)
{
int pos = query(sum[i]);
int tmp = sum[pos]^sum[i];
if(tmp > ans) ans = tmp,l = pos+1,r = i;
insert(sum[i],i);
}
printf("%d %d %d\n",ans,l,r);
return 0;
}