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的情况，你都要输出一个需要的步数，占一行。

---

LCT模板。

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back 
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
#define ls ch[x][1]
#define rs ch[x][0]
int rd()
{
    int sum = 0;char c = getchar();bool flag = true;
    while(c < '0' || c > '9') {if(c == '-') flag = false;c = getchar();}
    while(c >= '0' && c <= '9') sum = sum * 10 + c - 48,c = getchar();
    if(flag) return sum;
    else return -sum;
}
const int maxn = 2010000;
struct tree
{
    int ch[maxn][2],fa[maxn],v[maxn],mx[maxn],rev[maxn],sum[maxn];
    int q[maxn],top;
    void update(int x)
    {
        sum[x] = sum[ls] + sum[rs] + v[x];
        mx[x] = v[x];
        if(ls) mx[x] = max(mx[x],mx[ls]);
        if(rs) mx[x] = max(mx[x],mx[rs]);
    }
    bool isroot(int x){return ch[fa[x]][1] != x && ch[fa[x]][0] != x;}
    void pushdown(int x){if(rev[x])rev[x] ^= 1,swap(ls,rs),rev[ls] ^= 1,rev[rs] ^= 1;}
    void rorate(int x)
    {
        int y = fa[x],z = fa[y],k = ch[y][1] == x;
        if(!isroot(y)) ch[z][ch[z][1] == y] = x;
        ch[y][k] = ch[x][k^1]; fa[ch[x][k^1]] = y;
        ch[x][k^1] = y;        fa[x] = z;fa[y] = x;
        update(y);update(x);
    }
    void splay(int x)
    {
        int k = x;top = 0;
        while(1)
        {
            q[++top] = k;
            if(isroot(k)) break;
            k = fa[k];
        }
        while(top) pushdown(q[top--]);
        while(!isroot(x))
        {
            int y = fa[x],z = fa[y];
            if(!isroot(y)) rorate((ch[z][1]==y)^(ch[y][1] == x)?y:x);
            rorate(x);
        }
    }
    void access(int x){for(int t=0;x;x = fa[t=x]) splay(x),ch[x][1]=t,update(x);}
    void makeroot(int x){access(x);splay(x);rev[x] ^= 1;}
    void split(int x,int y){makeroot(x);access(y);splay(y);}
    void link(int x,int y){makeroot(x);fa[x] = y;}
    void cut(int x,int y)
    {
        makeroot(x);access(y);splay(y);
        fa[x] = ch[y][0] = 0;update(y);
    }
}T;
int n,m;
int a[201000];
int main()
{
    n = rd();
    rep(i,1,n)
    {
        T.v[i] = 1;
        a[i] = rd();
        if(a[i] + i > n) a[i] = n+1-i;
        T.link(i,a[i]+i);
    }
    m = rd();
    rep(i,1,m)
    {
        int k = rd();
        if(k == 1)
        {
            int x = rd()+1;
            T.split(x,n+1);
            printf("%d\n",T.sum[n+1]);
        }
        else
        {
            int x = rd()+1,v = rd();
            if(x + v > n) v = n+1-x;
            T.cut(x,x+a[x]);
            a[x] = v;
            T.link(x,x+a[x]);
        }
    }
    return 0;
}