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

来源：互联网发布：如何使用淘宝模板编辑：程序博客网时间：2024/04/26 18:05

某天，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 31 12 1 11 1

Sample Output

#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 400000using 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;}

1 0