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