CodeForces - 13E Holes

题面

题意

一条直线上有n个洞，每当一个球进入一个洞时，会被弹到后面距离它li的洞中或者直接被弹飞，要求支持两种操作：
1.修改某个洞的弹力
2.询问小球放入某个洞后，被弹几次之后会飞出，飞出之前进入的最后一个洞是什么。

做法

这题可以用LCT，但是码量大而且难写，虽然速度快一点。
最好还是用分块来写，对于每一个块维护其中每个洞弹出这个块要几次，会被弹到哪个洞，或者飞出。
询问时，一个一个块找下去即可。
修改时，暴力修改这个块内所有洞的值。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#define N 100100
using namespace std;

int n,m,num[N],s,cs[N],to[N],be[N],ans;

inline void get(int u)
{
    if(u+num[u]>n)
    {
        to[u]=-1;
        cs[u]=1;
    }
    else if(be[u]!=be[u+num[u]] || to[u+num[u]]==-1)
    {
        to[u]=u+num[u];
        cs[u]=1;
    }
    else
    {
        to[u]=to[u+num[u]];
        cs[u]=cs[u+num[u]]+1;
    }
}

int main()
{
    int i,j,o,p,q;
    cin>>n>>m;
    s=sqrt(n);
    for(i=1;i<=n;i++)
        be[i]=i/s+1;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
    }
    for(i=n;i>=1;i--)
        get(i);
    for(i=1;i<=m;i++)
    {
        scanf("%d",&o);
        if(!o)
        {
            scanf("%d%d",&p,&q);
            num[p]=q;
            for(j=p;be[j]==be[p];j--) get(j);
        }
        else
        {
            scanf("%d",&p);
            ans=0;
            for(;to[p]!=-1;p=to[p])
            {
                ans+=cs[p];
            }
            printf("%d %d\n",p,ans+1);
        }
    }
}