1. 程式人生 > 其它 >AcWing 1414. 牛異或(字首和+字典樹)

AcWing 1414. 牛異或(字首和+字典樹)

技術標籤:字典樹

題目連結:點選這裡

題目大意:
給出一個長度為 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; }