bzoj 2002 弹飞绵羊 分块（水）

这道题正解是LCT，http://blog.csdn.net/Loi_a/article/details/52692968
但我第一次是用分块水的。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
int a[200005];
int b[200005];
int c[200005];
int inf=214748;
int ask(int x)
{
    int ans=0;
    while(x!=inf)
    {
        ans+=b[x];
        x=c[x];
    }
    return ans;
}
int k,n;
void change(int j,int dq)
{
    if(dq==0)
    {
        dq=k;
        j-=1;
    }
    int x,ans;
    bool h=0;

    for(int i=dq;i>=1;i--)
    {
        x=j*k+i;ans=0;
        h=0;
        if(i==dq)
            while(x<=(j+1)*k&&x<=n)
            {
                x+=a[x];
                ans+=1;
            }
        else
        {
            x+=a[x];
            ans+=1;
            if(x<=(j+1)*k&&x<=n)
            {
                ans+=b[x];
                h=1;
            }
        }
        b[j*k+i]=ans;
        if(i==dq||h==0)
            c[j*k+i]=x;
        if(x>n)
            c[j*k+i]=inf;
        if(h==1)
        {
            if(c[x]>n)
                c[j*k+i]=inf;
            else
                c[j*k+i]=c[x];
        }
    }
}
int main()
{
    int m;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    k=sqrt(n);
    int x,y,z,ans;
    bool h=0;

    for(int j=0;j<k;j++)
    for(int i=k;i>=1;i--)
    {
        x=j*k+i;ans=0;h=0;
        while(x<=(j+1)*k&&!h)
        {
            x+=a[x];
            ans+=1;
            if(b[x])
            {
                ans+=b[x];
                h=1;
            }
        }
        b[j*k+i]=ans;
        if(x>n)
            c[j*k+i]=inf;
        if(h==1)
            c[j*k+i]=c[x];
        if(h==0&&x<=n)
            c[j*k+i]=x;
    }
    for(int i=n-k*k;i>=1;i--)
    {
        x=k*k+i;ans=0;h=0;
        while(x<=n&&!h)
        {
            x+=a[x];
            ans+=1;
            if(b[x])
            {
                ans+=b[x];
                h=1;
            }
        }
        b[k*k+i]=ans;
        if(x>n)
            c[k*k+i]=inf;
        if(h==1)
            c[k*k+i]=c[x];
        if(h==0&&x<=n)
            c[k*k+i]=x;
    }
    scanf("%d",&m);
    bool haha=0;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        y+=1;
        if(x==1)
        {
            if(haha==0)
                haha=1;
            else
                printf("\n");
            printf("%d",ask(y));
        }
        else
        {
            scanf("%d",&z);
            a[y]=z;
            change(y/k,y%k);
        }
    }
    return 0;
}