BZOJ 2002: [Hnoi2010]Bounce 弹飞绵羊（动态树LCT）

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 2843 Solved: 1519
[Submit][Status]

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

HINT

Source

Splay 启发式合并

动态树，形成一颗树。

改变树的边，然后维护子树大小就可以了

 /* ***********************************************

 Author        :kuangbin

 Created Time  :2013-9-4 7:41:21

 File Name     :BZOJ2002弹飞绵羊.cpp

 ************************************************ */

 #include <stdio.h>

 #include <string.h>

 #include <iostream>

 #include <algorithm>

 #include <vector>

 #include <queue>

 #include <set>

 #include <map>

 #include <string>

 #include <math.h>

 #include <stdlib.h>

 #include <time.h>

 using namespace std;

 const int MAXN = ;

 int ch[MAXN][],pre[MAXN];

 int size[MAXN];

 bool rt[MAXN];

 void push_down(int r)

 {

 }

 void push_up(int r)

 {

     size[r] = size[ch[r][]] + size[ch[r][]] + ;

 }

 void Rotate(int x)

 {

     int y = pre[x], kind = ch[y][]==x;

     ch[y][kind] = ch[x][!kind];

     pre[ch[y][kind]] = y;

     pre[x] = pre[y];

     pre[y] = x;

     ch[x][!kind] = y;

     if(rt[y])

         rt[y] = false, rt[x] = true;

     else

         ch[pre[x]][ch[pre[x]][]==y] = x;

     push_up(y);

 }

 void P(int r)

 {

     if(!rt[r])P(pre[r]);

     push_down(r);

 }

 void Splay(int r)

 {

     P(r);

     while( !rt[r] )

     {

         int f = pre[r], ff = pre[f];

         if(rt[f])

             Rotate(r);

         else if( (ch[ff][]==f)==(ch[f][]==r) )

             Rotate(f), Rotate(r);

         else

             Rotate(r), Rotate(r);

     }

     push_up(r);

 }

 int Access(int x)

 {

     int y = ;

     for( ; x ; x = pre[y=x])

     {

         Splay(x);

         rt[ch[x][]] = true, rt[ch[x][]=y] = false;

         push_up(x);

     }

     return y;

 }

 int a[MAXN];

 int main()

 {

     //freopen("in.txt","r",stdin);

     //freopen("out.txt","w",stdout);

     int n,m;

     while(scanf("%d",&n) == )

     {

         for(int i = ;i <= n;i++)

             scanf("%d",&a[i]);

         for(int i = ;i <= n+;i++)

         {

             pre[i] = ;

             ch[i][] = ch[i][] = ;

             rt[i] = true;

             size[i] = ;

         }

         size[] = ;

         for(int i = ;i <= n;i++)

         {

             int t = i + a[i];

             if( t > n+) t = n+;

             pre[i] = t;

         }

         int op;

         int u,v;

         scanf("%d",&m);

         while(m--)

         {

             scanf("%d",&op);

             if(op == )

             {

                 scanf("%d",&u);

                 u++;

                 Access(u);

                 Splay(u);

                 printf("%d\n",size[ch[u][]]);

             }

             else

             {

                 scanf("%d%d",&u,&v);

                 u++;

                 Access(u);

                 Splay(u);

                 pre[ch[u][]] = pre[u];

                 pre[u] = ;

                 rt[ch[u][]] = true;

                 ch[u][] = ;

                 push_up(u);

                 int t = u + v;

                 if(t > n+) t = n+;

                 pre[u] = t;

             }

         }

     }

     return ;

 }