【bzoj 2002】弹飞绵羊

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

唉隔壁都在弹飞大爷了，我还在弹绵羊>_<

分块大法【Time:1472 ms】

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 const int N=;
 int n,m,num,pl,w,bl,cnt;
 int k[N],be[N],st[N],to[N];
 int read()
 {
     int x=,f=;char c=getchar();
     while(c<''||c>''){if(c=='-')f=-;c=getchar();}
     while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
     return x*f;
 }
 int ask(int t)
 {
     int ans=st[t];
     while(to[t]<=n)
     {
         t=to[t];
         ans+=st[t];
     }
     return ans;
 }
 int main()
 {
     n=read();bl=sqrt(n);
     if(n%bl)cnt=n/bl+;
     else cnt=n/bl;
     for(int i=;i<=n;i++)
     {
         k[i]=read();
         be[i]=(i-)/bl+;
     }
     for(int i=n;i>;i--)
     {
         if(be[i]==be[i+k[i]]){st[i]=st[i+k[i]]+;to[i]=to[i+k[i]];}
         else {st[i]=;to[i]=i+k[i];}
     }
     m=read();
     while(m--)
     {
         num=read();pl=read()+;
         if(num==)printf("%d\n",ask(pl));
         else
         {
             w=read();
             k[pl]=w;
             int l=(be[pl]-)*bl+;
             for(int i=pl;i>=l;i--)
             {
                 if(be[i]==be[i+k[i]]){st[i]=st[i+k[i]]+;to[i]=to[i+k[i]];}
                 else {st[i]=;to[i]=i+k[i];}
             }
         }
     }
     return ;
 }

LCT【Time:1844 ms】

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 const int N=;
 int n,m,c[N][],next[N],fa[N],size[N],st[N];
 bool rev[N];
 int read()
 {
     int x=,f=;char c=getchar();
     while(c<''||c>''){if(c=='-')f=-;c=getchar();}
     while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
     return x*f;
 }
 bool isroot(int k){return c[fa[k]][]!=k&&c[fa[k]][]!=k;}
 void up(int x){size[x]=size[c[x][]]+size[c[x][]]+;}
 void down(int x)
 {
     int l=c[x][],r=c[x][];
     if(rev[x]){rev[x]^=;rev[l]^=;rev[r]^=;swap(c[x][],c[x][]);}
 }
 void rotate(int x)
 {
     int y=fa[x],z=fa[y],l,r;
     if(c[y][]==x)l=;else l=;r=l^;
     if(!isroot(y)){if(c[z][]==y)c[z][]=x;else c[z][]=x;}
     fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
     c[y][l]=c[x][r];c[x][r]=y;
     up(y);up(x);
 }
 void splay(int x)
 {
     int top=;st[++top]=x;
     for(int i=x;!isroot(i);i=fa[i])st[++top]=fa[i];
     for(int i=top;i;i--)down(st[i]);
     while(!isroot(x))
     {
         int y=fa[x],z=fa[y];
         if(!isroot(y))
         {
             if((c[y][]==x)^(c[z][]==y))rotate(x);
             else rotate(y);
         }
         rotate(x);
     }
 }
 void acs(int x)
 {
     int t=;
     while(x){splay(x);c[x][]=t;t=x;x=fa[x];}
 }
 void mkroot(int x)
 {
     acs(x);splay(x);rev[x]^=;
 }
 void link(int x,int y)
 {
     mkroot(x);fa[x]=y;splay(x);
 }
 void cut(int x,int y)
 {
     mkroot(x);acs(y);splay(y);c[y][]=fa[x]=;
 }
 int main()
 {
     n=read();
     for(int i=;i<=n;i++)
     {
         int x=read();
         fa[i]=x+i;size[i]=;
         if(fa[i]>n+)fa[i]=n+;
         next[i]=fa[i];
     }
     size[n+]=;
     m=read();
     while(m--)
     {
         int f=read();
         if(f==)
         {
             mkroot(n+);
             int x=read()+;
             acs(x);splay(x);printf("%d\n",size[c[x][]]);
         }
         else
         {
             int x=read()+,y=read();
             int t=min(n+,x+y);
             cut(x,next[x]);link(x,t);next[x]=t;
         }
     }
     return ;
 }