BZOJ2002弹飞绵羊

2002: [Hnoi2010]Bounce 弹飞绵羊
Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 6908 Solved: 3621
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
LCT裸题。。
直接把x和x+i link，然后最后求子树大小
修改的时候先cut再link
附上本蒟蒻的代码：

#include<cstdio>
#include<iostream>
using namespace std;
#define MAXN 200002
int n,m,father[MAXN],c[MAXN][2],st[MAXN],size[MAXN],next[MAXN];
bool rev[MAXN];

int read()
{
    int w=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') f=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*f;
}

bool isroot(int x)
{
    return c[father[x]][0]!=x && c[father[x]][1]!=x;
}

void update(int x)
{
    int l=c[x][0],r=c[x][1];
    size[x]=size[l]+size[r]+1;
}

void pushdown(int x)
{
    int l=c[x][0],r=c[x][1];
    if (rev[x]) rev[x]^=1,rev[l]^=1,rev[r]^=1,swap(c[x][0],c[x][1]);
}

void rotate(int x)
{
    int y=father[x],z=father[y],l,r;
    if (c[y][0]==x) l=0;
    else l=1;
    r=l^1;
    if (!isroot(y))
      if (c[z][0]==y) c[z][0]=x;
      else c[z][1]=x;
    father[x]=z,father[y]=x,father[c[x][r]]=y,c[y][l]=c[x][r],c[x][r]=y,update(y),update(x);
}

void splay(int x)
{
    int i,top=0,y,z;
    st[++top]=x;
    for (i=x;!isroot(i);i=father[i]) st[++top]=father[i];
    for (i=top;i;i--) pushdown(st[i]);
    while (!isroot(x))
      {
        y=father[x],z=father[y];
        if (!isroot(y))
          if (c[y][0]==x^c[z][0]==y) rotate(x);
          else rotate(y);
        rotate(x);
      }
}

void access(int x)
{
    int t=0;
    while (x)
      splay(x),c[x][1]=t,t=x,x=father[x];
}

void rever(int x)
{
    access(x),splay(x),rev[x]^=1;
}

void cut(int x,int y)
{
    rever(x),access(y),splay(y),c[y][0]=father[x]=0;
}

void link(int x,int y)
{
    rever(x),father[x]=y,splay(x);
}

int main()
{
    int i,x,p,y,t;
    n=read();
    for (i=1;i<=n;i++)
      {
        x=read(),father[i]=x+i,size[i]=1;
        if (father[i]>n+1) father[i]=n+1;
        next[i]=father[i];
      }
    size[n+1]=1,m=read();
    for (i=1;i<=m;i++)
      {
        p=read();
        if (p==1) rever(n+1),x=read(),x++,access(x),splay(x),printf("%d\n",size[c[x][0]]);
        if (p==2) x=read(),y=read(),x++,t=min(n+1,x+y),cut(x,next[x]),link(x,t),next[x]=t;
      }
    return 0;
}