bzoj2002 Bounce 弹飞绵羊 LCT

题目：某天，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的情况，你都要输出一个需要的步数，占一行。

思路：看成n个点，再加一个编号为n+1的点，跳出的点pre指向n+1，这样就成了一棵树，然后就可以套LCT了

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")#include<iostream>#include<algorithm>#include<cstdio>#include<cmath>#include<cstring>#include<string>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<list>#include<numeric>using namespace std;#define LL long long#define ULL unsigned long long#define INF 0x3f3f3f3f#define mm(a,b) memset(a,b,sizeof(a))#define PP puts("*********************");template<class T> T f_abs(T a){ return a > 0 ? a : -a; }template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}// 0x3f3f3f3f3f3f3f3f// 0x3f3f3f3fconst int maxn=2e5+50;const int maxe=2*maxn;struct Edge{    int to,next;}edge[maxe];int ch[maxn][2],key[maxn],pre[maxn],rev[maxn];int siz[maxn];bool rt[maxn];//标记节点是不是splay的根int n;//n个节点，从1开始计数void init(){    for(int i=0;i<=n+1;i++){        rt[i]=true;        ch[i][0]=ch[i][1]=0;        pre[i]=0;        rev[i]=0;        siz[i]=1;    }    siz[0]=0;}void update_rev(int x){    if(!x) return;    swap(ch[x][0],ch[x][1]);    rev[x]^=1;}void push_down(int x){    if(rev[x]){        update_rev(ch[x][0]);        update_rev(ch[x][1]);        rev[x]=0;    }}void push_up(int x){    siz[x]=1;    siz[x]+=siz[ch[x][0]];    siz[x]+=siz[ch[x][1]];}void rotate(int x){    int y=pre[x],d=ch[y][1]==x;    ch[y][d]=ch[x][!d];    pre[ch[y][d]]=y;    pre[x]=pre[y];    pre[y]=x;    ch[x][!d]=y;    if(rt[y]) rt[y]=false,rt[x]=true;    else ch[pre[x]][ch[pre[x]][1]==y]=x;    push_up(y);}//P函数先将根结点到x的路径上所有的结点的标记逐级下放void P(int x){    if(!rt[x]) P(pre[x]);    push_down(x);}//将x旋转到根void splay(int x){    P(x);    while(!rt[x]){        int f=pre[x],ff=pre[f];        if(rt[f]) rotate(x);        else if((ch[ff][1]==f)==(ch[f][1]==x))            rotate(f),rotate(x);        else            rotate(x),rotate(x);    }    push_up(x);}//将x到根的路径变成首选边int Access(int x){    int y=0;    for(;x;x=pre[y=x]){        splay(x);        rt[ch[x][1]]=true;        rt[ch[x][1]=y]=false;        push_up(x);    }    return y;}int arr[maxn];int main(){    int m,op,u,v;    while(~scanf("%d",&n)){        init();        for(int i=1;i<=n;i++)            scanf("%d",&arr[i]);        for(int i=1;i<=n;i++){            int t=i+arr[i];            if(t>n) t=n+1;            pre[i]=t;        }        scanf("%d",&m);        while(m--){            scanf("%d",&op);            if(op==1){                scanf("%d",&u);u++;                Access(u);                splay(u);                printf("%d\n",siz[ch[u][0]]);            }            else{                scanf("%d%d",&u,&v);u++;                Access(u);                splay(u);                pre[ch[u][0]]=0;                rt[ch[u][0]]=true;                ch[u][0]=0;                int t=u+v;                if(t>n) t=n+1;                pre[u]=t;            }        }    }    return 0;}