luogu3203弹飞绵羊

链接

　　https://www.luogu.org/problem/show?pid=3203

题解

　　我知道这个用LCT做看上去很裸，然而我不会哎
　　分块，每个点记录弹出第一次弹出这个块会到达哪个块，以及多少步弹出这个块。
　　查询的时候跳跃的单位是块，所以复杂度降为$\sqrt N$。修改的时候块与块之间不会影响，只把块内重算一次就好了，复杂度也算是$\sqrt N$
　　时间复杂度$O(N+M\sqrt N)$

代码

//分块
#include <cstdio>
#include <algorithm>
#define maxn 200010
#define size 450
using namespace std;
int N, M, k[maxn], to[maxn], lp[maxn], step[maxn];
void update(int x)
{
    if(lp[x+k[x]]!=lp[x] or x+k[x]>=N)to[x]=x+k[x],step[x]=1;
    else to[x]=to[x+k[x]],step[x]=step[x+k[x]]+1;
}
void init()
{
    int i;
    scanf("%d",&N);
    for(i=0;i<N;i++)scanf("%d",k+i),lp[i]=i/size;
    for(i=N-1;i>=0;i--)update(i);
}
int calc(int x)
{
    int ans=0;
    for(;x<N;x=to[x])ans+=step[x];
    return ans;
}
void solve()
{
    int type, a, b, i;
    scanf("%d",&M);
    while(M--)
    {
        scanf("%d%d",&type,&a);
        if(type==1)printf("%d\n",calc(a));
        else
        {
            scanf("%d",&b);
            k[a]=b;
            for(i=a;lp[i]==lp[a] and i>=0;i--)update(i);
        }
    }
}
int main()
{
    init();
    solve();
    return 0;
}