[HNOI 2010] 弹飞绵羊

题目描述：

有N个装置 每个装置有一个参数K
在i装置上可以到i+k位置
问几次可以弹飞出去

题目要求

要资瓷修改和查询操作
如果没修改用dfs即可
修改可咋办QWQ
我们考虑用LCT来维护这些点的连接关系
当i+k>n时 fa[i+k]=n+1
这样每次查询的时候即为查询X点到n+1的距离
这样我们每次让 n+1 makeroot
然后让access(x) splay(x)
由于Splay是以深度为关键字的 x为它所在的Splay里深度最大的
那x到n+1的距离，即为x左子树的节点数目

题目链接：

Luogu 3203
Bzoj 2002
Cogs 1689

Ac 代码：

#include <cstdio>
#include <iostream>
#define il inline
const int maxm=410000;
int n,m,net[maxm];
namespace LCT{
    int top,son[maxm][2],fa[maxm],siz[maxm],stk[maxm],rev[maxm];
    il void pushup(int x){siz[x]=siz[son[x][1]]+siz[son[x][0]]+1;}
    il void pushdown(int x)
    {
        if(rev[x])
        {
            rev[son[x][1]]^=1,rev[son[x][0]]^=1,rev[x]=0;
            std::swap(son[x][1],son[x][0]);
        }
    }
    il bool isroot(int x){return (son[fa[x]][1]!=x&&son[fa[x]][0]!=x);}
    void rotate(int x)
    {
        int fa1=fa[x],fa2=fa[fa1],l;
        if(son[fa1][0]==x) l=0;
        else l=1;
        int r=l^1;
        if(!isroot(fa1))
        {
         if(son[fa2][0]==fa1) son[fa2][0]=x;
         else son[fa2][1]=x;
        }
        fa[x]=fa2;fa[fa1]=x;fa[son[x][r]]=fa1;
        son[fa1][l]=son[x][r];son[x][r]=fa1;
        pushup(fa1);pushup(x);
    }
    void splay(int x)
    {
        top=1;stk[top]=x;
        for(int i=x;!isroot(i);i=fa[i]) stk[++top]=fa[i];
        for(int i=top;i;i--) pushdown(stk[i]);
        while(!isroot(x))
        {
            int y=fa[x],z=fa[y];
            if(!isroot(y))
            {
                if((son[y][0]==x)^(son[z][0]==y)) rotate(x);
                else rotate(y);
            }
            rotate(x);
        }
    }
    il void access(int x){for(int t=0;x;t=x,x=fa[x]) splay(x),son[x][1]=t,pushup(x);}//打通重路径 
    il void makeroot(int x){access(x),splay(x),rev[x]^=1;}//变为此Slpay的root 
    il int findroot(int x){access(x),splay(x);while(son[x][0])x=son[x][0];return x;}//寻找x所在树的根 
    void split(int x,int y){makeroot(x);access(y);splay(y);}
    void cut(int x,int y){split(x,y);if(son[y][0]==x) son[y][0]=0,fa[x]=0;}
    void link(int x,int y){makeroot(x);fa[x]=y;}
};
int main()
{
    //freopen("bzoj_2002.in","r",stdin);
    //freopen("bzoj_2002.out","w",stdout);
    scanf("%d",&n);
    LCT::siz[n+1]=1;
    for(int i=1,k;i<=n;i++)
    {
        scanf("%d",&k);
        LCT::siz[i]=1;
        LCT::fa[i]=i+k<=n?i+k:n+1;
        net[i]=LCT::fa[i];
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int opt,u,v;
        scanf("%d%d",&opt,&u);
        u++; 
        if(opt==1)
        {
            LCT::makeroot(n+1),LCT::access(u),LCT::splay(u);
            printf("%d\n",LCT::siz[LCT::son[u][0]]);
        }
        else
        {
            scanf("%d",&v);
            LCT::cut(u,net[u]),LCT::link(u,std::min(u+v,n+1)),net[u]=std::min(u+v,n+1);
        }
    }
    return 0;
}