bsoj 3175 【HNOI2010】弹飞绵羊

Description

　　Lostmonkey发明了一种超级反弹装置。
　　为了在绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿一条直线摆放 n个反弹装置，并按从前往后的方式将反弹装置依次编号为 0 到 n-1，对 0≤i≤n-1，为第 i 个反弹装置设定了初始弹力系数 ki，当绵羊落到第 i 个反弹装置上时，它将被往后弹出 ki 步，即落到第 i+ki 个反弹装置上，若不存在第i+ki个反弹装置，则绵羊被弹飞。
　　绵羊想知道： 从第i个反弹装置开始， 它被弹出几次 （含被弹飞的那次）后会被弹飞。为使游戏更有趣，Lostmonkey 还可以修改某个反弹装置的弹力系数，但任何时候弹力系数均为正整数。

Input

　　输入文件第一行是一个整数n，表示地上摆放n个反弹装置，
　　输入文件第二行是用空格隔开的n个正整数k0,k1,…,kn-1，分别表示n个反弹装置的初始弹力系数。
　　输入文件第三行是一个正整数m，表示后面还有m行输入数据。
　　接下来的m行，每行至少有用空格隔开的两个整数i和j，若i=1，则你要输出从第j个反弹装置开始，被弹出几次后会被弹飞；若i=2，则该行有用空格隔开的三个整数i，j和k，表示第j个反弹装置的弹力系数被修改为k。
　　输入的数据保证20%的数据满足n,m≤10000。100%的数据满足n≤200000,m≤100000

Output

　　输出文件output.txt包含的行数等于输入文件最后m行中i=1的行数。第h行输出一个整数，表示对输入中给出的第h个求弹出次数的问题，基于n个反弹装置当时的弹力系数，求出的弹出次数。

Sample Input

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

Sample Output

23

我的代码，不知道为什么一直Timeout。：

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
struct Link_cut_Tree{
	int fa;
	bool OnPath;
	int size;
	int l;
	int r;
	int rev;
}Tree[2000005];
int n;
int next[2000005];
int NEXT(int x,int y){
	if(x+y>n)return n+1;
	else return x+y;
}
void updata(int x){
Tree[x].size=1;
if(Tree[x].l)Tree[x].size+=Tree[Tree[x].l].size;
if(Tree[x].r)Tree[x].size+=Tree[Tree[x].r].size;}
void pushdown(int x){
	if(Tree[x].rev){
		swap(Tree[x].l,Tree[x].r);
		Tree[x].rev=0;
		Tree[Tree[x].l].rev^=1;
		Tree[Tree[x].r].rev^=1;
	}
}
void Rotate(int x){
	int y=Tree[x].fa;
	int z=Tree[y].fa;
	if(Tree[y].l==x){
		Tree[y].l=Tree[x].r;
		if(Tree[x].r)Tree[Tree[x].r].fa=y;
		if(Tree[y].OnPath){
			if(Tree[z].l==y)Tree[z].l=x;
			else Tree[z].r=x;
		}
		Tree[x].fa=z;
		Tree[x].r=y;
		Tree[y].fa=x;
		if(!Tree[y].OnPath){
			Tree[y].OnPath=1;
			Tree[x].OnPath=0;
		}
	}
	else{
		Tree[y].r=Tree[x].l;
		if(Tree[x].l)Tree[Tree[x].l].fa=y;
		if(Tree[y].OnPath){
			if(Tree[z].l==y)Tree[z].l=x;
			else Tree[z].r=x;
		}
		Tree[x].fa=z;
		Tree[x].l=y;
		Tree[y].fa=x;
		if(!Tree[y].OnPath){
			Tree[y].OnPath=1;
			Tree[x].OnPath=0;
		}
		
	}
	updata(y);
}
void Splay(int x){
	pushdown(x);
	while(Tree[x].OnPath){
		int y=Tree[x].fa;
		int z=Tree[y].fa; 
		if(Tree[y].OnPath)pushdown(z);
		pushdown(y);
		pushdown(x);
		if(!Tree[y].OnPath){
			Rotate(x);
		}
		else if((Tree[z].l==y)==(Tree[y].l==x)){
			Rotate(y);
			Rotate(x);
		}
		else {Rotate(x);Rotate(x);}
	}
	updata(x);
}
void ACCESS(int u){
	int v=0;
	while(u){
		Splay(u);
		Tree[Tree[u].r].OnPath=0;
		Tree[u].r=v;
		Tree[v].OnPath=1;
//		updata(u);
		v=u;
		u=Tree[u].fa;
		
	}
}
void Make_Root(int x){
	ACCESS(x);
	Splay(x);
	Tree[x].rev^=1;
}
void link(int x,int y){
	Make_Root(x);
	Tree[x].fa=y;
}
void make(int x){
	Make_Root(n+1);
	ACCESS(x);
	Splay(x);
}
void Cut(int x,int y){
	Make_Root(x);
	ACCESS(y);
	Splay(y);
	Tree[Tree[y].l].OnPath=Tree[y].l=Tree[x].fa=0;
}
int Getint(){
	char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	int ret=0;
	while(ch<='9'&&ch>='0')ret=ret*10+ch-'0',ch=getchar();
	return ret;
}
char bur[10];
void putint(int i){
	int p=0;
	if(i==0)p++;
	while(i){
		bur[p++]=i%10;
		i/=10;
	}
	for(int j=p-1;j>=0;j--)putchar('0'+bur[j]);
}
int main(){
//	freopen("a.in","r",stdin);
//	freopen("a.out","w",stdout);
	n=Getint();	int a,b;
	 
	for(int i=1;i<=n;i++){
		a=Getint();
		next[i]=min(i+a,n+1);
		link(i,next[i]);
	}
	int m;
	scanf("%d",&m);
	int t;
	for(int i=1;i<=m;i++){
		t=Getint();
		if(t==1){
			a=Getint();
			a++;
			make(a);
			putint(Tree[a].size-1);
			putchar('\n');
		}
		if(t==2){
			a=Getint();
			b=Getint();
			a++;
			int temp=min(n+1,a+b);
			if(next[a]!=temp){
				Cut(a,next[a]);
				next[a]=temp;
				link(a,next[a]);
			}
		}		
	}
	return 0;
}
/*
4
1 2 1 1
3
1 1
2 1 1
1 1



2 
3
*/

附一份AC代码：

http://blog.youkuaiyun.com/qq_30303087/article/details/50670728