【BZOJ2002】【HNOI2010】弹飞绵羊 [分块]

弹飞绵羊

Time Limit: 10 Sec  Memory Limit: 259 MB
[Submit][Status][Discuss]

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。

Output

　　对于每个i=1的情况，你都要输出一个需要的步数，占一行。

Sample Input

　　4
　　1 2 1 1
　　3
　　1 1
　　2 1 1
　　1 1

Sample Output

　　2
　　3

HINT

　　对于20%的数据：n,m<=10000;

　　对于100%的数据：n<=200000,m<=100000.

Main idea

　　给定一个位置，走一步可以走到一个指定位置，求走到n以外需要几步，走到的指定位置需要支持修改。

Solution

　　考虑暴力显然不可做，运用分块，维护两个信息：1.走出当前分块需要几步；2.走出当前分块后到了哪个点。

　　每次修改更新改点信息以及该块内的点，又因为一个点不一定只影响到该块内直接跳到这个点的点，也有间接关系的可能性存在，直接倒着搜一遍每次在可以跳到的点+1即可。

Code

 #include<iostream>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 using namespace std;  

 const int ONE=;

 int n,Q,P;
 int T,x,y;
 int num,to;
 int a[ONE],To[ONE],F[ONE];
 int go,jishu;
 int PD[ONE];
 int l[ONE],r[ONE];

 struct power
 {
         int step;
         int to;
         int from;
 }C[ONE];

 int get()
 {
         int res,Q=;    char c;
         while( (c=getchar())< || c>)
         if(c=='-')Q=-;
         if(Q) res=c-;
         while((c=getchar())>= && c<=)
         res=res*+c-;
         return res*Q;
 }

 int Query(int x)
 {
         int j=x;
         int Ans=;
         while(j<=n)
         {
             Ans+=C[j].step;
             j=C[j].to;
         }
         return Ans;
 }

 int Update(int x,int y)
 {
         int j=x;
         To[x]=x+y;
         if(To[x]>n) To[x]=n+;

         jishu=;
         while(C[j].from==C[x].from)
         {
             j=To[j];
             jishu++;
         }
         C[x].step=jishu; C[x].to=j;

         for(int i=x-;i>=l[C[x].from];i--)
         {
             int y=To[i];
             if(C[i].from==C[y].from)
             {
                 C[i].to=C[y].to;
                 C[i].step=C[y].step+;
             }
         }

 }

 int main()
 {
        // freopen("input.in","r",stdin);
        // freopen("input.out","w",stdout);
         n=get();
         Q=sqrt(n);
         for(int i=;i<=n;i++)
         {
             a[i]=get(); To[i]=i+a[i];

             if(To[i]>n) To[i]=n+;
             if(!((i-)%Q))
             {
                 num++;
                 l[num]=i;
             }
             C[i].from=num; r[num]=i;
         }

         for(int i=;i<=n;i++)
         {
             int j=i;
             jishu=;
             while(C[j].from==C[i].from)
             {
                 j=To[j];
                 jishu++;
             }
             C[i].step=jishu;
             C[i].to=j;
             if(C[i].to>n) C[i].to=n+;
         }

         T=get();
         while(T--)
         {
             P=get();
             if(P==)
             {
                 x=get(); x++;
                 printf("%d\n",Query(x));
             }

             if(P==)
             {
                 x=get(); x++;
                 y=get();
                 Update(x,y);
             }

         }
 }