BZOJ2002 [Hnoi2010]Bounce 弹飞绵羊

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

inline int in(){
    int x=0,f=1;char ch=getchar();
    while((ch>‘9‘ || ch<‘0‘) && ch!=‘-‘) ch=getchar();
    if(ch==‘-‘) ch=getchar(),f=-1;
    while(ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
    return x*f;
}

const int maxn=200010;

struct Node{
    int l,r,f,sz;
    bool rt;
}s[maxn];

int n,m;

void update(int x){
    s[x].sz=s[s[x].l].sz+s[s[x].r].sz+1;
}

void zig(int x){
    int y=s[x].f;
    s[x].f=s[y].f;
    if(s[y].rt) s[x].rt=true,s[y].rt=false;
    else{
        if(s[s[y].f].l==y) s[s[y].f].l=x;
        else s[s[y].f].r=x;
    }
    s[y].l=s[x].r;
    if(s[x].r) s[s[x].r].f=y;
    s[y].f=x,s[x].r=y;
    update(y);
}

void zag(int x){
    int y=s[x].f;
    s[x].f=s[y].f;
    if(s[y].rt) s[x].rt=true,s[y].rt=false;
    else{
        if(s[s[y].f].l==y) s[s[y].f].l=x;
        else s[s[y].f].r=x;
    }
    s[y].r=s[x].l;
    if(s[x].l) s[s[x].l].f=y;
    s[y].f=x,s[x].l=y;
    update(y);
}

void Splay(int x){
    int y;
    while(!s[x].rt){
        y=s[x].f;
        if(s[y].rt){ if(s[y].l==x) zig(x);
            else zag(x);}
        else{
            int z=s[y].f;
            if(y==s[z].l){ if(x==s[y].l) zig(y),zig(x);
                else zag(x),zig(x);}
            else{ if(x==s[y].r) zag(y),zag(x);
                else zig(x),zag(x);}
        }
    }
    update(x);
}

/*
    Access(x)表示的是将x与根节点相连的过程
    s[i].r[平衡树中的右节点]　在实际树中的意义是接在i后的点
    last表示的是现在x所在的链需要与之前x所在的链相连，也就需要将原本连接的部分舍去
    last初始为0，可以让x一开始便与后面的链断开
    所以操作就是一步步地往上爬，每次将上次的位置和这次的连上，直到根的过程
*/

void Access(int x){
    int last=0;
    for(;x;x=s[last=x].f){
        Splay(x);
        s[s[x].r].rt=true;
        s[x].r=last;
        s[last].rt=false;
        update(x);
    }
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("2002.in","r",stdin);
    freopen("2002.out","w",stdout);
#endif
    
    int t,ord,x,d;
    
    n=in();
    for(int i=1;i<=n;i++){
        t=in()+i;
        s[i].f=min(t,n+1);
        s[i].rt=true;
        s[i].sz=1;
    }
    s[0].rt=s[n+1].rt=true,s[n+1].sz=1;
    m=in();
    while(m--){
        ord=in();
        if(ord==1){
            x=in()+1;
            Access(x),Splay(x);
            printf("%d\n",s[s[x].l].sz);
        }
        else{
            x=in()+1;d=in();
            Access(x),Splay(x);
            s[s[x].l].f=s[x].f;
            s[s[x].l].rt=true;
            s[x].l=0,s[x].f=0;
            update(x);
            s[x].f=min(n+1,x+d);
        }
    }

    return 0;
}