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