bzoj 2002: [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

正解似乎是动态树还是啥的。。可是我不会啊！于是我就写了个分块。。

分成r块，每块长度n/r。每块内暴力维护跳几次可以跳出该块以及落点。

对于修改，只需要修改块内的，复杂度O(n/r)

对于询问，每块效率O(1)，复杂度O（r)

把r取sqrt(n)

那么复杂度就是msqrt(n)

#include<cmath>
#include<cstdio>
using namespace std;
int a[200001];
int x[300001][2];
int pl[200001],ps[200001];
int tot;
int n;
inline int ask(int d)
{
     int s=0;
     while(d<n)
     {
          s+=x[d][0];
          d=x[d][1];
     }
     return s;
}
inline void change(int s,int t)
{
	 a[s]=t;
     int i;
     for(i=s;i>=ps[pl[s]];i--)
     {
          if(i+a[i]>=ps[pl[i]+1])
          {
               x[i][0]=1;
               x[i][1]=i+a[i];
          }
          else
          {
               x[i][0]=x[i+a[i]][0]+1;
               x[i][1]=x[i+a[i]][1];
          }
     }
}
int main()
{
     scanf("%d",&n);
	 int r=int(sqrt(n));
     int i,j;
     j=r;
     for(i=0;i<=n-1;i++)
     {
          scanf("%d",&a[i]);
          if(j==r)
          {
               j=1;
               tot++;
               ps[tot]=i;
          }
          else
               j++;
          pl[i]=tot;
     }
     for(i=n-1;i>=0;i--)
     {
          if(i+a[i]>=ps[pl[i]+1])
          {
               x[i][0]=1;
               x[i][1]=i+a[i];
          }
          else
          {
               x[i][0]=x[i+a[i]][0]+1;
               x[i][1]=x[i+a[i]][1];
          }
     }
     int m;
     scanf("%d",&m);
     int k,s,t;
     for(i=1;i<=m;i++)
     {
          scanf("%d",&k);
          if(k==1)
          {
               scanf("%d",&s);
               printf("%d\n",ask(s));
          }
          else
          {
               scanf("%d%d",&s,&t);
               change(s,t);
          }
     }
     return 0;
}