BZOJ 3065 替罪羊树+动态开节点线段树
思路:
RT
可以看VFK的题解
我写了半天拍了半天...
不过是$nlog^2n$的
要写垃圾回收的
线段树 如果某个节点的sum是0 也可以free掉
//By SiriusRen #include <cstdio> #include <algorithm> using namespace std; const int N=70000; int root[200005],n,q,xx,yy,zz,Root,ans; struct Segment_Tree{ int rubbish[30000005],top,tree[30000005],lson[30000005],rson[30000005]; int get_rubbish(){if(rubbish[top])return rubbish[top--];return top--;} void insert(int l,int r,int &pos,int num,int wei){ if(!pos)pos=get_rubbish(); if(l==r){tree[pos]+=wei;return;} int mid=(l+r)>>1; if(mid>=num)insert(l,mid,lson[pos],num,wei); else insert(mid+1,r,rson[pos],num,wei); tree[pos]=tree[lson[pos]]+tree[rson[pos]]; if(!tree[lson[pos]]&&lson[pos])rubbish[++top]=lson[pos],lson[pos]=0; if(!tree[rson[pos]]&&rson[pos])rubbish[++top]=rson[pos],rson[pos]=0; } void merge(int x,int y,int &k){ if(!k)k=get_rubbish(); tree[k]=tree[x]+tree[y]; if(lson[x]||lson[y])merge(lson[x],lson[y],lson[k]); if(rson[x]||rson[y])merge(rson[x],rson[y],rson[k]); } void dfs(int l,int r,int x){ int mid=(l+r)>>1; if(lson[x])dfs(l,mid,lson[x]),rubbish[++top]=lson[x]; if(rson[x])dfs(mid+1,r,rson[x]),rubbish[++top]=rson[x]; lson[x]=rson[x]=tree[x]=0; } }segtree; int top1,top2,stk1[200005],stk2[200005]; struct SC_TREE{ int top,rubbish[200005],lson[200005],rson[200005],wei[200005],size[200005]; int get_rubbish(){return rubbish[top--];} void build(int &k,int l,int r){ if(l>r)return; if(!k)k=get_rubbish(); int mid=(l+r)>>1;wei[k]=stk2[mid]; if(l==r){segtree.insert(0,N,root[k],stk2[l],1),size[k]=1;return;} build(lson[k],l,mid-1),build(rson[k],mid+1,r); segtree.merge(root[lson[k]],root[rson[k]],root[k]); segtree.insert(0,N,root[k],stk2[mid],1); size[k]=size[lson[k]]+size[rson[k]]+1; } void init(){for(int i=1;i<=200000;i++)rubbish[i]=200000-i+1;top=200000;segtree.top=30000000,build(Root,1,n);} int find_wei(int x,int sz){ if(size[lson[x]]+1==sz)return wei[x]; else if(size[lson[x]]>=sz)return find_wei(lson[x],sz); else return find_wei(rson[x],sz-size[lson[x]]-1); } void change(int x,int sz,int w,int last_w){ segtree.insert(0,N,root[x],last_w,-1); segtree.insert(0,N,root[x],w,1); if(size[lson[x]]+1==sz){wei[x]=w;return;} else if(size[lson[x]]>=sz)change(lson[x],sz,w,last_w); else change(rson[x],sz-size[lson[x]]-1,w,last_w); } void query(int l,int r,int x){ int L=size[lson[x]],R=size[x]; if(l==1&&R==r){stk1[++top1]=root[x];return;} if(l<=L+1&&r>=L+1)stk2[++top2]=wei[x]; if(r<=L)query(l,r,lson[x]); else if(l>L+1)query(l-L-1,r-L-1,rson[x]); else{ if(l<=L)query(l,L,lson[x]); if(r>L+1)query(1,r-L-1,rson[x]); } } int Query(int l,int r,int kth){ top1=top2=0;query(l,r,Root); int L=0,R=N,ans=0,M,temp; while(L<R){ M=(L+R)>>1;temp=0; for(int i=1;i<=top1;i++)temp+=segtree.tree[segtree.lson[stk1[i]]]; for(int i=1;i<=top2;i++)if(stk2[i]<=M&&stk2[i]>=L)temp++; if(temp>kth){for(int i=1;i<=top1;i++)stk1[i]=segtree.lson[stk1[i]];R=M;} else {for(int i=1;i<=top1;i++)stk1[i]=segtree.rson[stk1[i]];L=M+1;kth-=temp;} }return L; } void dfs(int x){ segtree.dfs(0,N,root[x]);root[x]=0; if(lson[x])dfs(lson[x]),rubbish[++top]=lson[x]; stk2[++top2]=wei[x]; if(rson[x])dfs(rson[x]),rubbish[++top]=rson[x]; lson[x]=rson[x]=wei[x]=size[x]=0; } int rebuild(int sz,int &x){top2=0,dfs(x),build(x,1,sz);} bool insert(int &x,int sz,int w){ if(!x){x=get_rubbish();segtree.insert(0,N,root[x],w,1);size[x]=1;wei[x]=w;return 0;} segtree.insert(0,N,root[x],w,1),size[x]++; if(size[lson[x]]>=sz){if(insert(lson[x],sz,w))rebuild(size[lson[x]],lson[x]);} else if(insert(rson[x],sz-size[lson[x]]-1,w))rebuild(size[rson[x]],rson[x]); if(max(size[lson[x]],size[rson[x]])>0.666*size[x])return 1;return 0; } }SC_Tree; int read(){ int x=0;char p=getchar(); while(p<'0'||p>'9')p=getchar(); while(p>='0'&&p<='9')x=x*10+p-'0',p=getchar(); return x; } char readchar(){char p=getchar();while(p!='Q'&&p!='M'&&p!='I')p=getchar();return p;} int main(){ scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%d",&stk2[i]); SC_Tree.init(); scanf("%d",&q); while(q--){ char op=readchar();xx=read()^ans,yy=read()^ans; if(op=='Q')zz=read()^ans,printf("%d\n",ans=SC_Tree.Query(xx,yy,zz-1)); else if(op=='M')SC_Tree.change(Root,xx,yy,SC_Tree.find_wei(Root,xx)); else SC_Tree.insert(Root,xx-1,yy); } }