【分块】【LCT】bzoj2002 [Hnoi2010]Bounce 弹飞绵羊

分块，每个点统计还有几步弹出该块，以及它弹出块后的下一个节点是哪个点。

注意：update某个点的时候，会可能对当前块内 该点及以前的点 产生影响，所以对这部分点进行更新。

#include<cstdio>
#include<cmath>
using namespace std;
int n,m,op,x,y,sz,a[200001],l[450],sum,num[200001],b[200001],c[200001];
bool vis[200001];
void makeblock()
{
    int r;sz=sqrt(n);
    for(sum=1;sum*sz<n;sum++)
      {
        l[sum]=(sum-1)*sz;
        r=sum*sz-1;
          for(int i=l[sum];i<=r;i++)
          num[i]=sum;
      }
    l[sum]=sz*(sum-1);
    r=n-1;
    for(int i=l[sum];i<=r;i++)
      num[i]=sum;
}
void init()
{
    for(int i=0;i<n;i++)
      {
        int u=i,cnt=0;
        while(num[u]==num[i]){u+=a[u];cnt++;}
        b[i]=cnt;c[i]=u;
      }
}
inline int query(int u)
{
    int res=0;
    while(u<n){res+=b[u];u=c[u];}
    return res;
}
inline void update(const int &p,const int &val)
{
    a[p]=val;
    if(num[p+a[p]]!=num[p]){b[p]=1;c[p]=p+a[p];}
    else{b[p]=b[p+a[p]]+1;c[p]=c[p+a[p]];}
    for(int i=p;i>=l[num[p]];i--)
      if(i+a[i]<=p){b[i]=b[i+a[i]]+1;c[i]=c[i+a[i]];}
}
int Res,Num;char C,CH[20];
inline int G()
{
    Res=0;C='*'; 
    while(C<'0'||C>'9')C=getchar();
    while(C>='0'&&C<='9'){Res=Res*10+(C-'0');C=getchar();}
    return Res;
}
inline void P(int x)
{
    Num=0;  
    while(x>0)CH[++Num]=x%10,x/=10;
    while(Num)putchar(CH[Num--]+48);
    putchar('
');
}
int main()
{
    n=G();for(int i=0;i<n;i++)a[i]=G();
    makeblock();init();m=G();
    for(int i=1;i<=m;i++)
      {
          op=G();
          if(op==1){x=G();P(query(x));}
          else{x=G();y=G();update(x,y);}
      }
    return 0;
}