链接
https://www.luogu.org/problem/show?pid=3203
题解
我知道这个用LCT做看上去很裸,然而我不会哎
分块,每个点记录弹出第一次弹出这个块会到达哪个块,以及多少步弹出这个块。
查询的时候跳跃的单位是块,所以复杂度降为
N−−√
。修改的时候块与块之间不会影响,只把块内重算一次就好了,复杂度也算是
N−−√
时间复杂度
O(N+MN−−√)
代码
//分块
#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;
}