BZOJ3065(替罪羊树套线段树)

以前看到这题都瑟瑟发抖，终于过了心情舒畅。

按下标为关键字建替罪羊树，每个结点开一个权值线段树，维护的这个结点代表的子树的信息。

这题还得垃圾回收，自己yy的，不知对不对..

#include <cstdio>
#include <algorithm>
using namespace std;
#define M ((L+R)>>1)
#define l(x) t[x].s[0]
#define r(x) t[x].s[1]
#define L(x) t2[x].s[0]
#define R(x) t2[x].s[1]
 
const int N=;
char op[];
int n,x,y,z,q,rt,tt,t1,la,tp,tp2,V,st[N],st2[N],rb[N*],rb2[N*];
struct nd {int x,v,sz,s[];}t[N*],t2[N*];
 
void dl(int x) {if(!x) return; rb2[++tt]=x,dl(L(x)),dl(R(x)),t2[x].x=L(x)=R(x)=;}
void upd(int &o,int L,int R,int x,int y) {
    if(!o) o=rb2[tt--]; t2[o].x+=y;
    if(L==R) return;
    if(x<=M) upd(L(o),L,M,x,y); else upd(R(o),M+,R,x,y);
    if(!t2[o].x) dl(o),o=;
}
void ge(int x) {if(!x) return; ge(l(x)),st[++tp]=x,ge(r(x));}
void mr(int l,int r,int &x) {
    if(!l&&!r) return; t2[x=rb2[tt--]].x=t2[l].x+t2[r].x,mr(L(l),L(r),L(x)),mr(R(l),R(r),R(x));
}
void bd(int &x,int L,int R) {
    if(L>R) return; t[x=rb[t1--]].v=t[st[M]].v;
    if(L==R) t[x].sz=;
    else bd(l(x),L,M-),bd(r(x),M+,R),mr(t[l(x)].x,t[r(x)].x,t[x].x),t[x].sz=t[l(x)].sz+t[r(x)].sz+;
    upd(t[x].x,,N,t[x].v,);
}
void rm() {for(int i=;i<=tp;i++) dl(t[st[i]].x),l(st[i])=r(st[i])=t[st[i]].x=t[st[i]].v=t[st[i]].sz=;}
void ins(int &x,int p,int v) {
    if(!x) x=rb[t1--]; upd(t[x].x,,N,v,),t[x].sz++;
    if(!t[x].v) {t[x].v=v; return;}
    if(t[l(x)].sz+>=p) ins(l(x),p,v); else ins(r(x),p-t[l(x)].sz-,v);
    if(t[x].sz*0.75<max(t[l(x)].sz,t[r(x)].sz)) tp=,ge(x),bd(x,,tp),rm();
}
void gt(int x,int l,int r) {
    if(r-l+==t[x].sz) {st[++tp]=t[x].x; return;}
    if(l<=t[l(x)].sz) gt(l(x),l,min(r,t[l(x)].sz));
    if(l<=t[l(x)].sz+&&r>=t[l(x)].sz+) st2[++tp2]=t[x].v;
    if(r>t[l(x)].sz+) gt(r(x),max(l-t[l(x)].sz-,),r-t[l(x)].sz-);
}
int dfs(int L,int R,int k) {
    if(L==R) return L; int s=;
    for(int i=;i<=tp;i++) s+=t2[L(st[i])].x;
    for(int i=;i<=tp2;i++) if(st2[i]<=M&&st2[i]>=L) s++;
    if(s>=k) {
        for(int i=;i<=tp;i++) st[i]=L(st[i]);
        return dfs(L,M,k);
    } else {
        for(int i=;i<=tp;i++) st[i]=R(st[i]);
        return dfs(M+,R,k-s);
    }
}
int qr(int l,int r,int k) {tp=tp2=,gt(rt,l,r); return dfs(,N,k);}
void gai(int x,int p,int v) {
    if(t[l(x)].sz+==p) V=t[x].v,t[x].v=v; else if(t[l(x)].sz>=p) gai(l(x),p,v); else gai(r(x),p-t[l(x)].sz-,v);
    upd(t[x].x,,N,v,),upd(t[x].x,,N,V,-);
}
 
int main() {
    for(int i=;i<N*;i++) rb[i]=i;
    for(int i=;i<N*;i++) rb2[i]=i;
    scanf("%d",&n),tt=N*-,t1=N*-;
    for(int i=;i<=n;i++) scanf("%d",&x),ins(rt,i,x+);
    scanf("%d",&q);
    while(q--) {
        scanf("%s%d%d",op,&x,&y),x^=la,y^=la;
        if(op[]=='Q') scanf("%d",&z),printf("%d\n",la=qr(x,y,z^la)-);
        else if(op[]=='I') ins(rt,x,y+);
        else gai(rt,x,y+);
    }
    return ;
}