Bounce 弹飞绵羊 HYSBZ - 2002 分块

//预处理出以这个点为起点并跳出这个块的次数和位置
//更新一个点的弹力系数可以只更新这个点以及这个块内之前的点
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<cmath>
#include<iostream>
using namespace std;
const int N=2e5+;
struct node
{
    //ahead为从该点跳的距离
    int ahead;
    //step为跳到下一个块的所需步数
    int step;
    //Next为跳到的下一个块  跳出Next为0
    int Next;
    //pos为跳到下一个块的位置
    int pos;
}p[N];
int temp;
int block[N];
int main()
{
    int n,m,op;
    scanf("%d",&n);
    for(int i=; i<n; i++)
        scanf("%d",&p[i].ahead);
    temp=sqrt(n);
    //分块
    for(int i=; i<n; i++)
        block[i]=i/temp+;
    for(int i=n-; i>=; i--)
    {
        //如果在同一个块内
        if(block[i+p[i].ahead]!=block[i])
        {
            p[i].Next=block[i+p[i].ahead];
            p[i].pos=i+p[i].ahead;
            p[i].step=;
        }
        else
        {
            p[i].Next=p[i+p[i].ahead].Next;
            p[i].pos=p[i+p[i].ahead].pos;
            p[i].step=p[i+p[i].ahead].step+;
        }
    }
    scanf("%d",&m);
    int x,y;
    for(int cas=; cas<=m; ++cas)
    {
        scanf("%d",&op);
        if(op==)
        {
            scanf("%d",&x);
            int ans=;
            //一个块一个块的跳
            while(p[x].Next!=)
            {
                ans+=p[x].step;
                x=p[x].pos;
            }
            ans+=p[x].step;
            printf("%d\n",ans);
        }
        else if(op==)//更新
        {
            scanf("%d%d",&x,&y);
            p[x].ahead=y;
            //更新的话，只更新在同一块内的
            int l=(block[x]-)*temp;
            for(int i=x; i>=l; i--)
            {
                if(block[i+p[i].ahead]!=block[i])
                {
                    p[i].Next=block[i+p[i].ahead];
                    p[i].pos=i+p[i].ahead;
                    p[i].step=;
                }
                else
                {
                    p[i].Next=p[i+p[i].ahead].Next;
                    p[i].pos=p[i+p[i].ahead].pos;
                    p[i].step=p[i+p[i].ahead].step+;
                }
            }
        }
    }
    return ;
}