【bzoj2002】[Hnoi2010]Bounce 弹飞绵羊 link-cut-tree

2016-05-30 11:51:59

用一个next数组，记录点x的下一个点是哪个

查询时，moveroot(n+1),access(x),splay(x) ,输出size[ch[x][0]]即为答案

更改时，cut(x,next[x]) link(x,min(x+k,n+1))

记得splay旋转后要更新size

 #include<bits/stdc++.h>
 #define N 200005
 using namespace std;
 int read(){
   int x=,f=;char ch=getchar();
   while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
   while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
   return x*f;
 }
 int n,m,next[N];
 namespace LCT{
   int ch[N][],fa[N],sz[N];bool rev[N];
   inline bool isroot(int x){
     if(ch[fa[x]][]!=x&&ch[fa[x]][]!=x)return ;return ;
   }
   inline void pushdown(int x){
     if(rev[x]){
       rev[ch[x][]]^=;rev[ch[x][]]^=;rev[x]=;
       swap(ch[x][],ch[x][]);
     }
   }
   inline void pushup(int x){
     sz[x]=sz[ch[x][]]+sz[ch[x][]]+;
   }
   void rotate(int x){
     int y=fa[x],k=ch[y][]==x;
     if(!isroot(y))ch[fa[y]][ch[fa[y]][]==y]=x;
     fa[x]=fa[y];
     fa[ch[x][!k]]=y;
     fa[y]=x;
     ch[y][k]=ch[x][!k];
     ch[x][!k]=y;
     pushup(y);pushup(x);
   }
   void Pushdown(int x){
     if(!isroot(x))Pushdown(fa[x]);
     pushdown(x);
   }
   void splay(int x){
       Pushdown(x);
       for(int y=fa[x];!isroot(x);y=fa[x]){
         if(!isroot(y)){
             if((ch[fa[y]][ch[fa[y]][]==y])!=(ch[y][]==x))rotate(x);
             else rotate(y);
         }rotate(x);
     }
   }
   inline void access(int x){
     int y=;
     while(x){
       splay(x);
       ch[x][]=y;
       x=fa[y=x];
     }
   }
   inline void moveroot(int x){
     access(x);splay(x);rev[x]^=;
   }
   inline void link(int x,int y){
       moveroot(x);fa[x]=y;splay(x);
   }
   inline void cut(int x,int y){
       moveroot(x);access(y);splay(y);ch[y][]=fa[x]=;
   }
   inline int query(int x){
       moveroot(n+);access(x);splay(x);
       return sz[ch[x][]];
   }
 }
 int main(){
   n=read();
   for(int i=;i<=n;i++){
     int x=read();
     LCT::fa[i]=x+i;LCT::sz[i]=;
     if(LCT::fa[i]>n)LCT::fa[i]=n+;
     next[i]=LCT::fa[i];
   }
   m=read();
   while(m--){
     int t=read();
     if(t==)printf("%d\n",LCT::query(read()+));
     else {
       int x=read(),y=read();x++;
       int t=min(n+,x+y);
       LCT::cut(x,next[x]);LCT::link(x,t);next[x]=t;
     }
   }
   return ;
 }

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 7451  Solved: 3925
[Submit][Status][Discuss]

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