bzoj2002: [Hnoi2010]Bounce 弹飞绵羊 [分块][LCT]

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辣！)

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<cmath>
 using namespace std;

 inline int read(){
     char ch;
     int re=;
     bool flag=;
     while((ch=getchar())!='-'&&(ch<''||ch>''));
     ch=='-'?flag=:re=ch-'';
     while((ch=getchar())>=''&&ch<='')  re=re*+ch-'';
     return flag?-re:re;
 }

 const int maxn=;

 int n,m;
 //b[][1]跳到下一分块中的位置
 //b[][0]跳到下一分块需要的步数
 //c[]分块序号
 int a[maxn],b[maxn][],c[maxn];

 int main(){
     //freopen("temp.in","r",stdin);
     n=read();
     for(int i=;i<n;i++)  a[i]=read();
     for(int i=,v=,h=(int)sqrt(n);i<n;i++){
         c[i]=v;
         if((i+)%h==)  v++;
     }
     for(int i=n-;i>=;i--){
         //跳出分块或弹飞
         if(i+a[i]>n-||c[i+a[i]]!=c[i])  b[i][]=i+a[i];
         else  b[i][]=b[i+a[i]][];
         //记录多少步跳出分块
         if(c[i+a[i]]!=c[i])  b[i][]=;
         else  b[i][]=b[i+a[i]][]+;
     }
     m=read();
     int opt,pos,num,ans;
     while(m--){
         opt=read();
         if(opt==){
             pos=read();
             ans=;
             while(pos<=n-){
                 ans+=b[pos][];
                 pos=b[pos][];
             }
             if(ans==)  puts("");
             else  printf("%d\n",ans);
         }
         else{
             pos=read();  num=read();
             a[pos]=num;
             if(pos+num>n-||c[pos+num]!=c[pos])  b[pos][]=pos+num;
             else  b[pos][]=b[pos+num][];
             if(c[pos+num]!=c[pos])  b[pos][]=;
             else  b[pos][]=b[pos+num][]+;
             for(int i=pos-;i>=;i--){
                 if(c[i]!=c[pos])  break;
                 if(i+a[i]<=n-&&c[i+a[i]]==c[i])  b[i][]=b[i+a[i]][];
                 if(c[i+a[i]]==c[i])  b[i][]=b[i+a[i]][]+;
             }
         }
     }
     return ;
 }

---