【bzoj2002】[Hnoi2010]Bounce 弹飞绵羊

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

题解
把每个作为他弹向的点的儿子，然后lct维护，答案就是深度。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define N 200005
int n,m;
int a[N],c[N][2],fa[N],size[N],st[N];
bool rev[N];
using namespace std;
int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9'){ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}

bool isroot(int x)
{
    return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;
}

void update(int x)
{
    size[x]=size[c[x][0]]+size[c[x][1]]+1;
}

void pushdown(int x)
{
    int l=c[x][0],r=c[x][1];
    if(rev[x])
    {
        rev[x]^=1;rev[l]^=1;rev[r]^=1;
        swap(c[x][0],c[x][1]);
    }
}

void rotate(int x)
{
    int l,r,y=fa[x],z=fa[y];
    if (c[y][0]==x) l=0;else l=1;r=l^1;
    if (!isroot(y))
    {
        if (c[z][0]==y) c[z][0]=x;else c[z][1]=x;
    }
    fa[x]=z;fa[y]=x;
    fa[c[x][r]]=y;
    c[y][l]=c[x][r];
    c[x][r]=y;
    update(y);update(x);
}

void splay(int x)
{
    int top=0;
    st[++top]=x;
    for (int i=x;!isroot(i);i=fa[i])
    {
        st[++top]=fa[i];
    }
    for (int i=top;i;i--) pushdown(st[i]);
    while (!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if (!isroot(y))
        {
            if (c[y][0]==x^c[z][0]==y) rotate(x);else rotate(y);
        }
        rotate(x);
    }
}

void access(int x)
{
    int t=0;
    while (x)
    {
        splay(x);
        c[x][1]=t;
        t=x;
        x=fa[x];
    }
}

void link(int x,int y)
{
    access(x);
    splay(x);
    rev[x]^=1;
    fa[x]=y;
    splay(x);
}

void cut(int x)
{
    access(x);
    splay(x);
    fa[c[x][0]]=0;
    c[x][0]=0;
    update(x);
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++)
    {
        a[i]=read();
        link(i,min(n+1,i+a[i]));
    }
    m=read();
    for (int i=1;i<=m;i++)
    {
        int opt=read(),x=read()+1;
        if (opt==1)
        {
            access(x);
            splay(x);
            printf("%d\n",size[c[x][0]]);
        }
        else
        {
            cut(x);
            a[x]=read();
            link(x,min(n+1,x+a[x]));
        }
    }
    return 0;
}