BZOJ 2002 弹飞绵羊 分块

题意:有n个装置编号0~n-1,每个装置有弹力k[i],在第i个装置给以弹到第i+k[i]个装置(若i+k[i]不存在 游戏结束)
现在有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;
}