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