P3203 [HNOI2010]弹飞绵羊

题目链接:点我

题目大意:有n个装置，每个装置设定初始弹力系数ki,当达到第i个装置时,会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则被弹飞。求从第i个装置起步时，被弹几次后会被弹飞。带修改操作,下标0开始

分析:
开始打算倒着跑一遍记录每个位置弹飞要多少次,但是发现这样做修改操作复杂度爆炸.
为了减少修改复杂度我们可以用分块的思想,将其分成$\sqrt{n}$块,修改时只要对一个块进行操作.

下面是代码:

#include <bits/stdc++.h>
#define in(b) scanf("%d",&b)

using namespace std;
const int N=3e5+10;
int num,bel[N],l[N],r[N],len,a[N],to[N],step[N];
void update(int i)//修改块
{
    for(int j=r[i];j>=l[i];j--)
    {
        if(j+a[j]>r[i]) to[j]=j+a[j],step[j]=1;
        else
        {
            to[j]=to[j+a[j]];
            step[j]=1+step[j+a[j]];
        }
    }
}
void build(int n)//构造块
{
    len=sqrt(n);num=n/len;
    if(n%len) num++;
    for(int i=1;i<=num;i++)
    l[i]=(i-1)*len+1,r[i]=i*len;
    r[num]=n;
    for(int i=1;i<=n;i++)
    {
        bel[i]=(i-1)/len+1;
    }
    for(int i=1;i<=num;i++)
    {
        update(i);
    }
}
int query(int x,int n)//查询答案
{
    int ans=0;
    while(x<=n)
    {
        // debug(x)
        ans+=step[x];
        x=to[x];
    }
    return ans;
}
int main()
{
    int n;in(n);
    for(int i=1;i<=n;i++) in(a[i]);
    build(n);
    int m;in(m);
    for(int i=1;i<=m;i++)
    {
        int op;in(op);
        if(op==1)
        {
            int x;in(x);printf("%d\n",query(x+1,n));
        }
        else
        {
            int x,v;in(x),in(v);
            x+=1;
            a[x]=v;
            update(bel[x]);
        }
    }
    return 0;
}
/*

*/