bzoj 2002 Bounce 弹飞绵羊(LCT动态树)

某天，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

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define F(x) tree[x].fa
#define LC(x) tree[x].child[0]
#define RC(x) tree[x].child[1]
#define REV(x) tree[x].rev
#define SIZE(x) tree[x].size
#define Size 400000
using namespace std;
inline int read(){
	int sum=0,fg=1;char c=getchar();
	while(c<'0' || c>'9'){if(c=='-')fg=-1;c=getchar();}
	while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}
	return sum*fg;
}
struct lct{
	int fa,child[2],rev,size;
}tree[Size];
int fa[Size];
struct link_cut_tree{
	int stack[Size];
	inline bool isroot(int x){
		return LC(F(x))!=x && RC(F(x))!=x;
	}
	inline void pushup(int x){
		SIZE(x)=SIZE(LC(x))+SIZE(RC(x))+1;
	}
	inline void pushdown(int x){
		if(!REV(x))return;
		REV(x)^=1;REV(LC(x))^=1;REV(RC(x))^=1;
		swap(LC(x),RC(x));
	}
	inline void rotate(int x){
		int A=F(x),B=F(A);bool w=(RC(A)==x);
		if(!isroot(A)){
			if(LC(B)==A)LC(B)=x;
			else if(RC(B)==A)RC(B)=x;
		}
		F(tree[x].child[w^1])=A;F(A)=x;F(x)=B;
		tree[A].child[w]=tree[x].child[w^1];tree[x].child[w^1]=A;
		pushup(A);pushup(x);
	}
	void Pushdown(int x){
		if(!isroot(x))Pushdown(F(x));
		pushdown(x);
	}
	inline void splay(int x){
		Pushdown(x); 
		while(!isroot(x)){
			if(!isroot(F(x)))rotate(x);
			rotate(x);
		}
	}
	inline void access(int x){
		for(int i=0;x;x=F(x)){splay(x);RC(x)=i;i=x;pushup(x);}
	}
	inline void reverse(int x){
		access(x);splay(x);REV(x)^=1;
	}
	inline void link(int x,int y){
		reverse(x);F(x)=y;
	}
	inline void cut(int x,int y){
		reverse(x);access(y);splay(y);F(x)=LC(y)=0;
	}
}LCT; 
int main(){
	int n=read();
	for(int i=1;i<=n;i++){
		int k=read();
		SIZE(i)=1;fa[i]=F(i)=min(n+1,i+k);
	}
	SIZE(n+1)=1;
	int m=read();
	while(m--){
		int tp=read();
		if(tp==1){
			int x=read();x++;
			LCT.reverse(n+1);LCT.access(x);LCT.splay(x);
			printf("%d\n",SIZE(LC(x)));
		}
		else{
			int x=read(),y=read();x++;
			LCT.cut(x,fa[x]);LCT.link(x,min(n+1,x+y));
			fa[x]=min(n+1,x+y);
		}
	}
	return 0;
}