题意:有n个装置编号0~n-1,每个装置有弹力k[i],在第i个装置给以弹到第i+k[i]个装置(若i+k[i]不存在 游戏结束)
现在有m个操作,操作1:从第x个装置出发,弹多少次后游戏结束?,操作2:修改第i个装置弹力k[i]
n,m<=2e5.
序列分成根号n块,每个装置维护需要跳多少次到下一块及保存它在下一块的落点.
现在有m个操作,操作1:从第x个装置出发,弹多少次后游戏结束?,操作2:修改第i个装置弹力k[i]
n,m<=2e5.
序列分成根号n块,每个装置维护需要跳多少次到下一块及保存它在下一块的落点.
查询一次只要sqrt(n),更新时只有[L,x]的pos及tim可能发生变化,L为x所在块的左端点.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int n,block,k[N];
int pos[N],tim[N];
void build()
{
block=(int)sqrt(n);
for(int i=n-1;i>=0;i--)
{
int nxt=i+k[i];
int L=(i/block+1)*block;
if(nxt>=n)
pos[i]=-1,tim[i]=1;
else if(nxt>=L)
pos[i]=nxt,tim[i]=1;
else
pos[i]=pos[nxt],tim[i]=tim[nxt]+1;
}
}
int main()
{
while(cin>>n)
{
for(int i=0;i<n;i++)
scanf("%d",&k[i]);
build();
int Q,op,x,y;
cin>>Q;
while(Q--)
{
scanf("%d",&op);
if(op==1)
{
scanf("%d",&x);
int cnt=0;
while(x!=-1)
{
cnt+=tim[x];
x=pos[x];
}
printf("%d\n",cnt);
}
else
{
scanf("%d%d",&x,&y);
k[x]=y;
int l=x/block*block;//¸Ã¿é×ó¶Ëµã.
for(int i=x;i>=l;i--)
{
int nxt=i+k[i];
int L=(i/block+1)*block;
if(nxt>=n)
pos[i]=-1,tim[i]=1;
else if(nxt>=L)
pos[i]=nxt,tim[i]=1;
else
pos[i]=pos[nxt],tim[i]=tim[nxt]+1;
}
}
}
}
return 0;
}