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.csdn.net/qq_30303087/article/details/50670728