Bounce(弹走绵羊)lct裸题

Bounce（弹走绵羊）

某天，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<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 200000+11;
int size[maxn],child[maxn][2],fa[maxn];
void pushup(int x){
	size[x]=size[child[x][0]]+size[child[x][1]]+1;
}
bool isroot(int x){
	int y=fa[x];
	return (child[y][0]!=x&&child[y][1]!=x);
}
void _rot(int x){
	int y=fa[x],w=(child[y][1]==x);
	int g=fa[y];
	if(!isroot(y)){
		if(child[g][0]==y)child[g][0]=x;
		else if(child[g][1]==y)child[g][1]=x;
	}
	child[y][w]=child[x][w^1];
	fa[child[x][w^1]]=y;
	child[x][w^1]=y;
	fa[x]=g;
	fa[y]=x;	
	pushup(y);
}
void splay(int x){
	while(!isroot(x)){
		int y=fa[x],g=fa[y];
		if(isroot(y)){
			_rot(x);
			break;
		}
		if(!isroot(g))_rot(y);
		_rot(x);
	}
	pushup(x);
}
void accuss(int x){
	int y=0;
	while(x){
		splay(x);
		child[x][1]=y;
		pushup(x);
		y=x;
		x=fa[x];
	}
}
void cut(int x){
	accuss(x);
	splay(x);
	int l=child[x][0];
	fa[l]=0;
	child[x][0]=0;
	pushup(x);
}
void link(int x,int y){
	cut(x);
	fa[x]=y;
}
int main(){
	int i,j,k,m,n;
	scanf("%d",&n);
	for(i=1;i<=n;i++)size[i]=1;
	for(i=1;i<=n;i++){
		int val;
		scanf("%d",&val);
		if(i+val>n)fa[i]=0;
		else fa[i]=i+val;
	}
	size[0]=0;
	scanf("%d",&m);
	for(i=1;i<=m;i++){
		int t;
		scanf("%d",&t);
		if(t==1){
			int x;
			scanf("%d",&x);
			x++;
			accuss(x);
			splay(x);
			printf("%d\n",size[child[x][0]]+1);
		}
		else{
			int x,y;
			scanf("%d%d",&x,&y);
			x++;
			if(x+y>n)link(x,0);
			else link(x,x+y);
		}
	}
	return 0;
}