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 ;
 }