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的练手题，就是用LCT维护子树大小，不知道为什么就是比别人长了一倍（hzwer大神加起来都只要100行+，用时也比我快。。。）

AC code：

#include <ctime>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

inline int read()
{
    int x = 0; int v = 1, c = getchar();
    for(; c < '0' || c > '9'; c = getchar()) if(c == '-') v = -1;
    for(;c >= '0' && c <= '9'; c = getchar()) x = x * 10 + c - '0';
    return x * v;
}

template<typename T>
inline void write(T x, int ch = 10)
{
    int s[20], t = 0;
    if(x < 0) { x = -x; putchar('-'); }
    do s[t++] = x % 10; while(x /= 10);
    while(t--) putchar('0' + s[t]);
    putchar(ch);
}

const int maxn = 200005;

struct lct
{
    int lc[maxn], rc[maxn], fa[maxn], size[maxn];
    bool rev[maxn];

    inline bool isroot(const int& o)
    {
        return lc[fa[o]] != o && rc[fa[o]] != o;
    }

    inline void maintain(const int& o)
    {
        size[o] = size[lc[o]] + size[rc[o]] + 1;
    }

    inline void pushdown(const int& o)
    {
        if(rev[o])
        {
            rev[o] = 0; rev[lc[o]] ^= 1; rev[rc[o]] ^= 1;
            swap(lc[o], rc[o]);
        }
    }

    inline void setc(const int& x, const int& y, bool kind)
    {
        //make y to be the son of x
        !kind ? lc[x] = y : rc[x] = y;
        if(y) fa[y] = x;
    }

    inline void rotate(const int& x)
    {
        int p = fa[x];
        bool kind = rc[p] == x;
        if(!isroot(p))
            setc(fa[p], x, rc[fa[p]] == p);
        else
            fa[x] = fa[p];
        setc(p, !kind ? rc[x] : lc[x], kind);
        setc(x, p, kind^1);
        maintain(p);
        maintain(x);
    }

    inline void splay(const int& x)
    {
        static int s[maxn];

        int t = 0;

        s[t++] = x;
        for(int i = x; !isroot(i); i = fa[i])
            s[t++] = fa[i];

        while(t--)
            pushdown(s[t]);

        while(!isroot(x))
        {
            int y = fa[x], z = fa[y];
            if(!isroot(y))
            {
                if(lc[z] == y ^ lc[y] == x)
                    rotate(x);
                else
                    rotate(y);
            }
            rotate(x);
        }
    }

    inline void access(int x)
    {
        for(int c = 0; x; c = x, x = fa[x])
        {
            splay(x);
            rc[x] = c;
        }
    }

    inline void makeroot(const int& x)
    {
        access(x);
        splay(x);
        rev[x] ^= 1;
    }

    inline void link(const int& x, const int& y)
    {
        makeroot(x);
        fa[x] = y;
        splay(x);
    }

    inline void cut(const int& x, const int& y)
    {
        makeroot(x);
        access(y);
        splay(y);
        lc[y] = fa[x] = 0;
    }
};

int n, m;
int next[maxn];
lct T;

void input()
{
    n = read();
    for(int i = 1; i <= n; ++i)
    {
        int x = read();
        T.fa[i] = x + i;
        T.size[i] = 1;
        if(T.fa[i] > n + 1)
            T.fa[i] = n + 1;
        next[i] = T.fa[i];
    }
    T.size[n+1] = 1;
}

void calc(const int& x)
{
    T.makeroot(n + 1);
    T.access(x);
    T.splay(x);
    write(T.size[T.lc[x]]);
}

void modify(const int& x, const int& y)
{
    T.cut(x, next[x]);
    T.link(x, y);
    next[x] = y;
}

void solve()
{
    register int x;

    m = read();
    for(int i = 1; i <= m; ++i)
    {
        int op = read();
        x = read() + 1;
        if(op == 1)
            calc(x);
        else
            modify(x, min(n+1, x+read()));
    }
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif

    input();
    solve();

#ifndef ONLINE_JUDGE
    fprintf(stderr, "Run Time: %dms\n", clock());
    fclose(stdin), fclose(stdout);
#endif
    return 0;
}