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;
}