【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;
}

猜你喜欢

转载自www.cnblogs.com/cjyyb/p/10694026.html