【BZOJ 2002】弹飞绵羊

Description

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

Input

第一行包含一个整数n，表示地上有n个装置，装置的编号从0到n-1,接下来一行有n个正整数，依次为那n个装置的初始弹力系数。第三行有一个正整数m，接下来m行每行至少有两个数i、j，若i=1，你要输出从j出发被弹几次后被弹飞，若i=2则还会再输入一个正整数k，表示第j个弹力装置的系数被修改成k。对于20%的数据n,m<=10000，对于100%的数据n<=200000,m<=100000

Output

对于每个i=1的情况，你都要输出一个需要的步数，占一行。

Sample Input

4
1 2 1 1
3
1 1
2 1 1
1 1

Sample Output

2
3

思路

分块

对每个元素维护两个信息：
1、跳出其所在块需要的步数
2、跳出其所在块后跳到哪个点
很显然有了这两个信息可以根号时间内进行询问。
修改就暴力重构所在块。

当然可以用LCT

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
inline ll read()
{
	ll x=0,f=1;char ch=getchar();
	while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,block,cnt;
int pt[200005],st[200005],K[200005],belong[200005];
int l[1005],r[1005];
inline int cal(int x)
{
	int tmp=0;
	while(1)
	{
		tmp+=st[x];
		if(!pt[x])break;
		x=pt[x];
	}
	return tmp;
}
int main()
{
	n=read();block=sqrt(n);
	for(int i=1;i<=n;i++)
	    K[i]=read();
	if(n%block)cnt=n/block+1;
	else cnt=n/block;
	for(int i=1;i<=cnt;i++)
	    l[i]=(i-1)*block+1,r[i]=i*block;
	r[cnt]=n;
	for(int i=1;i<=n;i++)
	   belong[i]=(i-1)/block+1;
	for(int i=n;i>0;i--)
	{
		if(i+K[i]>n)st[i]=1;
		else if(belong[i]==belong[i+K[i]])
		    st[i]=st[i+K[i]]+1,pt[i]=pt[i+K[i]];
		else st[i]=1,pt[i]=i+K[i];
	}
	m=read();
	for(int i=1;i<=m;i++)
	{
		int f=read(),x=read(),y;
		x++;
		if(f==1)printf("%d\n",cal(x));
		else 
		{
			y=read();K[x]=y;
			for(int i=x;i>=l[belong[x]];i--)
			    if(belong[i]==belong[i+K[i]])
		            st[i]=st[i+K[i]]+1,pt[i]=pt[i+K[i]];
		        else st[i]=1,pt[i]=i+K[i];
		}
	}
	return 0;
}