BZOJ 2002: [Hnoi2010]Bounce 弹飞绵羊 LCT

Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 11244 Solved: 5719

Description

某天，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

3

1 1

2 1 1

1 1

Sample Output

2

3

HINT

Source

---

裸LCT，如果可以跳到一个点，那就认爸爸

---

#include<cstdio>#include<iostream>#include<algorithm>using namespace std;const int MAXN=200000+10;struct Tree{    int ch[2],size,fa;}tree[MAXN];int n,m,temp,kk,opt;void pushup(int x){    tree[x].size=1;    if(tree[x].ch[0]) tree[x].size+=tree[tree[x].ch[0]].size;    if(tree[x].ch[1]) tree[x].size+=tree[tree[x].ch[1]].size; } bool isroot(int x){    return tree[tree[x].fa].ch[0]!=x&&tree[tree[x].fa].ch[1]!=x;}void rotate(int x){    int y=tree[x].fa,z=tree[y].fa,l,r;    if(tree[y].ch[0]==x)l=0;else l=1;r=l^1;    if(!isroot(y)){        if(tree[z].ch[0]==y)tree[z].ch[0]=x;        else                tree[z].ch[1]=x;    }    tree[x].fa=z;tree[tree[x].ch[r]].fa=y;tree[y].fa=x;    tree[y].ch[l]=tree[x].ch[r];tree[x].ch[r]=y;    pushup(y);pushup(x);}void splay(int x){    while(!isroot(x)){        int y=tree[x].fa,z=tree[y].fa;        if(!isroot(y)){            if(tree[y].ch[0]==x^tree[z].ch[0]==y)rotate(x);            else                                 rotate(y);        }        rotate(x);    }}void access(int x){for(register int i=0;x;i=x,x=tree[x].fa){splay(x);tree[x].ch[1]=i;pushup(x);}}void link(int u,int v){    if(v>n)v=0;    access(u);    splay(u);    tree[tree[u].ch[0]].fa=0;    tree[u].ch[0]=0;    tree[u].fa=v;    pushup(u);}int main(){    scanf("%d",&n);    for(register int i=1;i<=n;i++){        tree[i].size = 1;        tree[i].fa=tree[i].ch[0]=tree[i].ch[1]=0;        scanf("%d",&temp);        link(i,i+temp);    }    tree[0].size=0;    scanf("%d",&m);    for(register int i=1;i<=m;i++){        scanf("%d",&opt);        if(opt==1){            scanf("%d",&temp);            temp++;            access(temp);splay(temp);            printf("%d\n",tree[tree[temp].ch[0]].size+1);        }else{            scanf("%d",&temp);temp++;scanf("%d",&kk);            link(temp,temp+kk);        }    }    return 0;}