首页 > 代码库 > BZOJ4966 : 总统选举
BZOJ4966 : 总统选举
线段树维护每个点的最有可能是答案的数以及它的权重。
合并两个节点的时候,将权重互相抵消,保留较大的那一个。
得到答案后,再在对应权值的Treap中查询出现次数,检查是否真正是答案。
时间复杂度$O(n\log n)$。
#include<cstdio>#include<cstdlib>const int N=500010,M=1100010,BUF=30000000;int n,m,i,a[N],c,d,pos[N],v[M],f[M],V,F;char Buf[BUF],*buf=Buf;inline void read(int&a){for(a=0;*buf<48;buf++);while(*buf>47)a=a*10+*buf++-48;}inline void merge(int a,int b,int c,int d,int&V,int&F){ if(b==d){V=a+c,F=b;return;} if(!b||!d){V=a+c,F=b+d;return;} if(a==c){V=F=0;return;} if(a>c){V=a-c,F=b;return;} V=c-a,F=d;}void build(int x,int a,int b){ if(a==b){ v[x]=1; f[x]=::a[a]; pos[a]=x; return; } int mid=(a+b)>>1; build(x<<1,a,mid),build(x<<1|1,mid+1,b); merge(v[x<<1],f[x<<1],v[x<<1|1],f[x<<1|1],v[x],f[x]);}struct node{ int val,cnt,sum,p;node*l,*r; node(){val=cnt=sum=p=0;l=r=NULL;} inline void up(){sum=cnt+l->sum+r->sum;}}*blank=new(node),pool[1500010],*cur=pool,*T[N];inline void Rotatel(node*&x){node*y=x->r;x->r=y->l;x->up();y->l=x;y->up();x=y;}inline void Rotater(node*&x){node*y=x->l;x->l=y->r;x->up();y->r=x;y->up();x=y;}void Insert(node*&x,int p){ if(x==blank){ x=cur++;x->val=p;x->l=x->r=blank;x->cnt=x->sum=1;x->p=rand(); return; } x->sum++; if(p==x->val){x->cnt++;return;} if(p<x->val){ Insert(x->l,p); if(x->l->p>x->p)Rotater(x); }else{ Insert(x->r,p); if(x->r->p>x->p)Rotatel(x); }}void Delete(node*&x,int p){ x->sum--; if(p==x->val){x->cnt--;return;} if(p<x->val)Delete(x->l,p);else Delete(x->r,p);}int Ask(node*&x,int p){ if(x==blank)return 0; if(p==x->val)return x->r->sum; if(p<x->val)return x->cnt+x->r->sum+Ask(x->l,p); return Ask(x->r,p);}inline void change(int x){ Delete(T[a[x]],x); Insert(T[a[x]=F],x); f[x=pos[x]]=F; for(x>>=1;x;x>>=1)merge(v[x<<1],f[x<<1],v[x<<1|1],f[x<<1|1],v[x],f[x]);}void ask(int x,int a,int b){ if(c<=a&&b<=d){ merge(V,F,v[x],f[x],V,F); return; } int mid=(a+b)>>1; if(c<=mid)ask(x<<1,a,mid); if(d>mid)ask(x<<1|1,mid+1,b);}inline bool check(int l,int r,int t){ if(!t)return 0; int w=Ask(T[t],l-1)-Ask(T[t],r); return w*2>r-l+1;}int main(){ fread(Buf,1,BUF,stdin);read(n),read(m); blank->l=blank->r=blank; for(i=1;i<=n;i++)T[i]=blank; for(i=1;i<=n;i++)read(a[i]),Insert(T[a[i]],i); build(1,1,n); while(m--){ read(c),read(d); V=F=0; ask(1,1,n); if(!check(c,d,F))F=0; read(c); if(!F)F=c; read(c); while(c--)read(d),change(d); printf("%d\n",F); } if(!check(1,n,f[1]))f[1]=-1; return printf("%d",f[1]),0;}
BZOJ4966 : 总统选举
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。