弹飞绵羊 bzoj 2002 分块

题目

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

分析

表示不会动态树，于是用了个分块。。。
这题的分块思想很巧妙。。。

将原序列分成n−√块，ei表示当前i跳出i所在的那一块需要的步数，eni表示跳出块后的位置。
执行询问操作时，只需要不断地跳出下一个块，跳到不能再跳为止，最多跳n−√次。
执行修改操作时，只需修改本块中在该数之前的ei和eni，从前往后修改，最多修改n−√次。
初始化的方法跟在执行修改操作时的更新方法差不多，
分块算法，奥妙重重！

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector> 

using namespace std;  

const int maxn=200000+100;
int e[maxn];
int en[maxn];
int belong[maxn];
int jump[maxn];

int block,cnt;
int l[maxn],r[maxn];

int n,m;

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&jump[i]);
    }
    block=sqrt(n);
    cnt=n/block;
    if (n%block) cnt++;
    for (int i=1;i<=cnt;i++)
    {
        l[i]=(i-1)*block+1; r[i]=i*block;
    }
    r[cnt]=n;
    int j=cnt;
    for (int i=n;i>0;i--)
    {
        if (i<l[j]) j--;
        belong[i]=j;    
        if (i+jump[i]>r[belong[i]])
        {
            e[i]=i+jump[i];
            en[i]=1;
        }
        else
        {
            e[i]=e[i+jump[i]];
            en[i]=en[i+jump[i]]+1;
        }
    }

    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        int c;
        scanf("%d",&c);
        if (c==1)
        {
            int x;
            scanf("%d",&x);
            x++;
            int ans=0;
            while (x<=n)
            {
                ans+=en[x];
                x=e[x];
            }
            printf("%d\n",ans);
        }
        else
        {
            int x,y;
            scanf("%d%d",&x,&y);
            x++;
            jump[x]=y;
            for (int i=x;i>=l[belong[x]];i--)
            {
                if (i+jump[i]>r[belong[i]])
                {
                    e[i]=i+jump[i];
                    en[i]=1;
                }
                else
                {
                    e[i]=e[i+jump[i]];
                    en[i]=en[i+jump[i]]+1;
                }
            }
        }
    }
}