[LCT] BZOJ2002: [Hnoi2010]Bounce 弹飞绵羊

题意

一条直线摆上n个装置，每个装置有个弹力系数ki，当绵羊达到第i个装置时，它会往后弹到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。
绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。
你需要执行m次操作，有两种类型：
1.修改某装置的弹力系数（ki始终为正整数）
2.询问从i起步被弹几次后会被弹飞
n,m<=200000

题解

算是LCT模板题吧。
添加一个点表示“弹飞”,那么n+1个点，n条边，显然是一颗树。
现在我们需要树上加/删边，且询问两点距离的数据结构。直接LCT即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const LL maxn=200005;
struct node{
    LL fa,ch[2],size;
    bool rev;
} T[maxn];
void maintain(int x){ T[x].size=T[T[x].ch[0]].size+T[T[x].ch[1]].size+1; }
bool is_root(int x){ return T[T[x].fa].ch[0]!=x&&T[T[x].fa].ch[1]!=x; }
void pushdown(int x){
    if(!T[x].rev) return;
    swap(T[x].ch[0],T[x].ch[1]);
    T[T[x].ch[0]].rev^=1; T[T[x].ch[1]].rev^=1; T[x].rev=0;
}
void rot(int k){
    int p=T[k].fa, d=(T[p].ch[1]==k)^1, _fa=T[p].fa;
    if(!is_root(p)) T[_fa].ch[T[_fa].ch[1]==p]=k;
    T[p].ch[d^1]=T[k].ch[d]; T[T[k].ch[d]].fa=p;
    T[k].ch[d]=p; T[p].fa=k;  
    maintain(p); maintain(k); T[k].fa=_fa;
}
int stk[maxn],top;
void Splay(int x){
    stk[++top]=x;
    for (int i=x;!is_root(i);i=T[i].fa) stk[++top]=T[i].fa;
    while(top) pushdown(stk[top--]);
    while(!is_root(x)){
        int y=T[x].fa, z=T[y].fa, d1=T[y].ch[1]==x, d2=T[z].ch[1]==y;
        if(!is_root(y)){
            if(d1==d2) rot(y), rot(x);
                  else rot(x), rot(x);
        } else rot(x);
    }
}
inline void Access(int x){
    for(int t=0;x;t=x,x=T[x].fa) Splay(x), T[x].ch[1]=t, maintain(x);
}
inline void make_root(int x){
    Access(x); Splay(x); T[x].rev^=1;
}
inline void Link(int x,int y){
    make_root(x); T[x].fa=y;
}
inline void Cut(int x,int y){
    make_root(x); Access(y); Splay(y);
    T[y].ch[0]=T[x].fa=0;
}
int n,Q,tan[maxn];
int main(){
    freopen("bzoj2002.in","r",stdin);
    freopen("bzoj2002.out","w",stdout);
    scanf("%d",&n); n++;
    for(int i=1;i<=n;i++) T[i].size=1;
    for(int i=1;i<=n-1;i++) scanf("%d",&tan[i]), Link(i,min(n,i+tan[i]));
    scanf("%d",&Q);
    while(Q--){
        int pd; scanf("%d",&pd); 
        if(pd==1){
            int x; scanf("%d",&x); x++;
            make_root(n); Access(x);
            Splay(x); printf("%d\n",T[x].size-1);
        } else{
            int x,y; scanf("%d%d",&x,&y); x++;
            Cut(x,min(n,x+tan[x])); Link(x,min(n,x+(tan[x]=y)));
        }
    }
    return 0;
}