【BZOJ5495】[十二省联考2019]异或粽子（主席树，贪心）

【BZOJ5495】[十二省联考2019]异或粽子（主席树，贪心）

题面

BZOJ
洛谷

题解

这不是送分题吗。。。
转异或前缀和，构建可持久化\(Trie\)。
然后拿一个堆维护每次的最大值，每次如果取了一个数，就把它再在\(Trie\)树上查一次新建一个元素丢回堆里就行了。

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
#define ll long long
#define MAX 500500
inline ll read()
{
	ll x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
struct Node{int ch[2],v;}t[MAX*80];
int tot,rt[MAX];
void Modify(int &x,int d,ll val,int w)
{
	++tot;t[tot]=t[x];x=tot;t[x].v+=w;if(d==-1)return;
	if((val>>d)&1)Modify(t[x].ch[1],d-1,val,w);
	else Modify(t[x].ch[0],d-1,val,w);
}
ll Query(int x,int d,ll val)
{
	if(d==-1)return 0;int c=(val>>d)&1;
	if(t[t[x].ch[c^1]].v)return (1ll<<d)+Query(t[x].ch[c^1],d-1,val);
	else return Query(t[x].ch[c],d-1,val);
}
struct Data{ll w;int x;};
bool operator<(Data a,Data b){return a.w<b.w;}	
priority_queue<Data> Q;
ll a[MAX],ans;int n,K;
int main()
{
	n=read();K=read();
	for(int i=1;i<=n;++i)a[i]=a[i-1]^read();
	for(int i=1;i<=n;++i)rt[i]=rt[i-1],Modify(rt[i],32,a[i-1],1);
	for(int i=1;i<=n;++i)Q.push((Data){Query(rt[i],32,a[i]),i});
	while(K--)
	{
		Data u=Q.top();Q.pop();
		ans+=u.w;Modify(rt[u.x],32,u.w^a[u.x],-1);
		Q.push((Data){Query(rt[u.x],32,a[u.x]),u.x});
	}
	printf("%lld\n",ans);
	return 0;
}