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

 

这个题给了10s,但是也不能乱搞.

我们现将弹簧进行分块,对于每个弹簧我们记录它跳出自己所在块的步数times跟跳出当前块的落点pos

这样我们对于每次更改就可以只更改x所在块中位于x之前的点,跟x自己,这样是每个询问是

然而对于每一个询问由于我们记录的pos是跳出当前块的落点,这样最多我们跳块,所以也是

 

综合复杂度

写完这个题感觉分块真是优美

 #include <bits/stdc++.h>

 using namespace std;

 const int maxn = ;

 int n,m;

 int q,op;

 int a[maxn];

 int pos[maxn],times[maxn];

 int main()

 {

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

     while (~scanf("%d",&n)){

         memset(pos,-,sizeof pos);

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

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

         }

         int sz = (int) sqrt(n);

         for (int i=n-;i>=;--i){//从后向前初始化pos,times

             int tmp = a[i]+i;

             if (tmp>n)

                 pos[i]=-,times[i]=;//直接跳出边界

             else if (tmp>=(i/sz+)*sz)

                 pos[i] = tmp,times[i]=;//跳出了当前块

             else

                 pos[i] = pos[tmp],times[i]=times[tmp]+;//跳到了块内的另一个点

         }

         scanf("%d",&m);

         for (int i=;i<m;++i){

             scanf("%d",&q);

             if (q==){

                 int x;

                 int ans = ;

                 scanf("%d",&x);

                 for (int j=x;j!=-;j=pos[j]){

                     ans+=times[j];

                 }

                 printf("%d\n",ans);

             }

             else{

                 int x,y;

                 scanf("%d%d",&x,&y);

                 a[x]=y;

                 for (int j=x;j>=x/sz*sz;j--){//更新只更新x所在块里面位于x之前的块跟x自己

                     int tmp=j+a[j];

                     if (tmp>=n)

                         pos[j]=-,times[j]=;

                     else if (tmp>=(j/sz+)*sz)

                         pos[j]=tmp,times[j]=;

                     else

                         pos[j]=pos[tmp],times[j] = times[tmp]+;

                 }

             }

         }

     }

     return ;

 }