bzoj2002: [Hnoi2010]Bounce 弹飞绵羊（lct）

Description

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

Input

第一行包含一个整数n，表示地上有n个装置，装置的编号从0到n-1,接下来一行有n个正整数，依次为那n个装置的初始弹力系数。第三行有一个正整数m，接下来m行每行至少有两个数i、j，若i=1，你要输出从j出发被弹几次后被弹飞，若i=2则还会再输入一个正整数k，表示第j个弹力装置的系数被修改成k。对于20%的数据n,m<=10000，对于100%的数据n<=200000,m<=100000

Output

对于每个i=1的情况，你都要输出一个需要的步数，占一行。

Sample Input

4
1 2 1 1
3
1 1
2 1 1
1 1

Sample Output
2
3

题解：动态树。
将每个点的父亲置为能弹到的点，注意是单向边，link()时不用makeroot()。
询问相当于求深度，直接access(),splay()输出size即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
const int Maxn=2e5+50;
inline int read()
{
    char ch=getchar();int i=0,f=1;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){i=(i<<1)+(i<<3)+ch-'0';ch=getchar();}
    return i*f;
}

int n,Q,k[Maxn];
struct node
{
    node *lc,*rc,*fa;
    int sze;
    node():lc(NULL),rc(NULL),fa(NULL),sze(1){}
    inline void upt()
    {
        sze=((lc?lc->sze:0)+(rc?rc->sze:0)+1); 
    }
}*tr[Maxn],Pool[Maxn],*pool=Pool;
inline bool isroot(node *x)
{
    return !x->fa||(x->fa->lc!=x&&x->fa->rc!=x);
}
inline void rotate(node *x)
{
    node *y=x->fa,*z=y->fa;
    if(z&&!isroot(y))
    {
        if(z->lc==y)z->lc=x;
        else z->rc=x;
    } 
    if(x==y->lc)
    {
        node *b=x->rc;
        if(b)b->fa=y;
        x->rc=y;
        y->lc=b;
    }
    else
    {
        node *b=x->lc;
        if(b)b->fa=y;
        x->lc=y;
        y->rc=b;
    }
    x->fa=z,y->fa=x;
    y->upt(),x->upt();
}
inline bool which(node *x)
{
    return x->fa->lc==x;
}
inline void splay(node *x)
{
    while(!isroot(x))
    {
        if(!isroot(x->fa))
        {
            if(which(x->fa)^which(x))rotate(x);
            else rotate(x->fa);
        }
        rotate(x);
    }
}
inline void access(node *x)
{
    for(node *y=NULL;x;y=x,x=x->fa)
    {
        splay(x),x->rc=y;
        if(y)y->fa=x;
        x->upt();
    }
}
inline void cut(node *x,node *y)
{
    access(x);
    splay(x);
    x->lc->fa=NULL;
    x->lc=NULL;
    x->upt();
}
inline void link(node *x,node *y)
{
    access(x);
    splay(x);
    x->fa=y;
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        tr[i]=++pool;
    } 
    for(int i=1;i<=n;i++)
    {
        k[i]=read();
        if(i+k[i]<=n)link(tr[i],tr[i+k[i]]);
        else continue; 
    }
    Q=read();
    while(Q--)
    {
        int x=read();
        if(x==1)
        {
            int y=read();
            y++;
            if(y>n)puts("0");
            else
            {
                access(tr[y]);
                splay(tr[y]);
                printf("%d\n",tr[y]->sze);
            }
        }
        else
        {
            int a=read(),b=read();
            a++;
            if(a>n)continue;
            if(a+k[a]<=n)cut(tr[a],tr[a+k[a]]); 
            if(a+b<=n)link(tr[a],tr[a+b]);
            k[a]=b;
        }
    }
}