Luogu5283 十二省联考2019异或粽子(trie/可持久化trie+堆)
做前缀异或和,用堆维护一个五元组(x,l,r,p,v),x为区间右端点的值,l~r为区间左端点的范围,p为x在l~r中最大异或和的位置,v为该最大异或和,每次从堆中取出v最大的元素,以p为界将其切成两部分重新扔进堆即可。查询一个值在一个区间中的最大异或和用可持久化trie实现。luogu上T掉了。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> #include<queue> using namespace std; #define ll long long #define N 500010 #define ui unsigned int char getc(){char c=getchar();while ((c<'A'||c>'Z')&&(c<'a'||c>'z')&&(c<'0'||c>'9')) c=getchar();return c;} int gcd(int n,int m){return m==0?n:gcd(m,n%m);} ui read() { ui x=0;char c=getchar(); while (c<'0'||c>'9') c=getchar(); while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x; } int n,k,goat,root[N],cnt; const ui I=1; ui a[N]; struct data{int ch[2],s,id; }trie[N*33]; void ins(int &k,ui x,int i,int ID) { trie[++cnt]=trie[k];k=cnt; trie[k].s++;trie[k].id=ID; if (i<0) return; ins(trie[k].ch[(x&(I<<i))>0],x,i-1,ID); } struct data2 { ui x;int l,r,p;ui d; bool operator <(const data2&a) const { return d<a.d; } }; priority_queue<data2> q; int query(int l,int r,ui x) { for (int i=31;~i;i--) { int op=(x&(I<<i))>0; if (trie[trie[r].ch[op^1]].s>trie[trie[l].ch[op^1]].s) l=trie[l].ch[op^1],r=trie[r].ch[op^1]; else l=trie[l].ch[op],r=trie[r].ch[op]; } return trie[r].id; } int main() { #ifndef ONLINE_JUDGE freopen("a.in","r",stdin); freopen("a.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif cin>>n>>k;ins(root[0],0,31,0); for (int i=1;i<=n;i++) { root[i]=root[i-1]; a[i]=a[i-1]^read(); ins(root[i],a[i],31,i); } root[-1]=0; for (int i=1;i<=n;i++) { int p=query(root[-1],root[i-1],a[i]); q.push((data2){a[i],0,i-1,p,a[i]^a[p]}); } ll ans=0; while (k--) { data2 t=q.top();q.pop();ans+=t.d;int x=t.p; if (x>t.l) { int p=query(root[t.l-1],root[x-1],t.x); q.push((data2){t.x,t.l,x-1,p,t.x^a[p]}); } if (x<t.r) { int p=query(root[x],root[t.r],t.x); q.push((data2){t.x,x+1,t.r,p,t.x^a[p]}); } } cout<<ans; return 0; }
场上写的是CF1055F的做法(当然没有隐式建trie了)。考场代码:
#include<bits/stdc++.h> using namespace std; #define ll long long #define N 500010 #define uint unsigned int #define mp(x,y) make_pair((x),(y)) uint read() { uint x=0;char c=getchar(); while (c<'0'||c>'9') c=getchar(); while (c>='0'&&c<='9') x=x*10+(c^48),c=getchar(); return x; } typedef pair<int,int> pii; typedef pair<uint,int> pui; int n,k,trie[N*33][2],size[N*33],nxt[N*33],head[N*33],T,cnt; pii p[N*4],q[N*4]; uint val[N*33],a[N],v[N*33]; pui b[N*4]; ll ans; void ins(uint x) { int k=0;uint I=1; for (int i=31;~i;i--) { if (!trie[k][(x&(I<<i))>0]) trie[k][(x&(I<<i))>0]=++cnt; k=trie[k][(x&(I<<i))>0];size[k]++; v[++T]=x;nxt[T]=head[k];head[k]=T; } val[k]=x; } bool cmp(pui a,pui b) { return a.first>b.first; } int main() { freopen("xor.in","r",stdin); freopen("xor.out","w",stdout); n=read(),k=read(); for (int i=1;i<=n;i++) a[i]=read(); for (int i=1;i<=n;i++) a[i]^=a[i-1]; for (int i=0;i<=n;i++) ins(a[i]); cnt=1;p[1]=mp(0,0); for (int _=32;_;_--) { ll tot=0; for (int i=1;i<=cnt;i++) { tot+=1ll*size[trie[p[i].first][0]]*size[trie[p[i].second][1]]; if (p[i].first!=p[i].second) tot+=1ll*size[trie[p[i].first][1]]*size[trie[p[i].second][0]]; } if (tot>k) { int u=0; for (int i=1;i<=cnt;i++) { if (p[i].first!=p[i].second&&(trie[p[i].first][0]>0&&trie[p[i].second][1]>0)) { q[++u].first=trie[p[i].first][0]; q[u].second=trie[p[i].second][1]; } if (trie[p[i].first][1]>0&&trie[p[i].second][0]>0) { q[++u].first=trie[p[i].first][1]; q[u].second=trie[p[i].second][0]; } } cnt=u;for (int i=1;i<=cnt;i++) p[i]=q[i]; } else { k-=tot; for (int i=1;i<=cnt;i++) { int L=trie[p[i].first][0],R=trie[p[i].second][1]; for (int x=head[L];x;x=nxt[x]) for (int y=head[R];y;y=nxt[y]) ans+=v[x]^v[y]; if (p[i].first!=p[i].second) { int L=trie[p[i].first][1],R=trie[p[i].second][0]; for (int x=head[L];x;x=nxt[x]) for (int y=head[R];y;y=nxt[y]) ans+=v[x]^v[y]; } } int u=0; for (int i=1;i<=cnt;i++) { if (trie[p[i].first][0]>0&&trie[p[i].second][0]>0) { q[++u].first=trie[p[i].first][0]; q[u].second=trie[p[i].second][0]; } if (trie[p[i].first][1]>0&&trie[p[i].second][1]>0) { q[++u].first=trie[p[i].first][1]; q[u].second=trie[p[i].second][1]; } } cnt=u;for (int i=1;i<=cnt;i++) p[i]=q[i]; } } for (int i=1;i<=cnt;i++) { b[i].first=val[p[i].first]^val[p[i].second]; if (1ll*size[p[i].first]*size[p[i].second]>k) b[i].second=k; else b[i].second=size[p[i].first]*size[p[i].second]; } sort(b+1,b+cnt+1,cmp); for (int i=1;i<=cnt;i++) if (k>b[i].second) ans+=1ll*b[i].first*b[i].second,k-=b[i].second; else {ans+=1ll*b[i].first*k;break;} cout<<ans; return 0; }