Bounce 弹飞绵羊（splay | 分块）

原题链接
Bounce 弹飞绵羊

Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 8285 Solved: 4266
[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
这个题其实是一个标准的LCT或者是伸展树的问题，但是用分块来解会更加方便。关于块状链表，就是把一个长为n数组分为√n份，把每一份作为一个相对独立的单元记算。也就是分治法的思想。对于此题，我们用s[]表示距离；belong[]表示此点属于哪一块；nex[]表示此点的后继；k[]表示此点的劲度系数。对于同一连通块内的两点，如果前一点可以跳到后一点，那么前一点的w[]应等于后一点的w[]，而前一点的s[]应为后一点的s[]+1。如果是不同连通块里的两点，前一点可以到达后一点，那么前一点的w[]指向后一点，前一点的s[]值为1。求值时累加。为什么分块后会更快，分块之后，通过把块内能到达的部分都进行了整合，同时指向了一个块内的祖先，这样原本打算指向块内的第一步，现在就会直接走动这一块的祖先上去，中间节省了很多步，所以得到了优化。当然跨块的时候那一步就没有办法避免了。但是如果不分块，一个位置可能就是由无数个位置走过来的，但是中间的重复计算却得不到有效的简化

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200000 + 10;
int n,m,cmd;
int k[maxn],nex[maxn],s[maxn],belong[maxn];
//k代表弹性系数，next代表该元素的后继，s表示中间的距离，belong的值表示属于哪一个块
void make(int i){
        int t=i+k[i];
        if(t>=n) t=n;//到了n就是边界了
        if(belong[i]==belong[t]){
                nex[i]=nex[t];//相同块内的后继都是一样的
                s[i]=s[t]+1;//这个距离是倒过来算的，从后面往前面算距离
        }
        else{
                nex[i]=t;//跨块的时候就是直接将后继指向下一个块
                s[i]=1;//跨块的时候的距离就是1了，这个距离后面总要加上
        }
}
int main(){
        while(scanf("%d",&n)==1){
                for(int i=0;i<n;i++)
                        scanf("%d",&k[i]);
                int mod=(int)sqrt(n);
                for(int i=(n-1);i>=0;i--)
                        belong[i]= (i/mod);//将每个元素所在的块进行分类
                for(int i=(n-1);i>=0;i--)
                        make(i);//对该元素进行分块处理
                scanf("%d",&m);
                while(m--){
                        scanf("%d",&cmd);
                        if(cmd==1){
                                int b,ans=0;
                                scanf("%d",&b);
                                for(int i=b;i<n;i=nex[i])
                                        ans+=s[i];//只需要不断地将这些距离进行相加即可
                                printf("%d\n",ans);
                        }
                        else{
                                int loc,t;
                                scanf("%d%d",&loc,&t);
                                k[loc]=t;
                                for(int i=loc;i>=0;i--){
                                        if(belong[i]==belong[loc]) make(i);
                                        else break;
                                }
                        }
                }
        }
        return 0;
}