BZOJ2002 [Hnoi2010]Bounce 弹飞绵羊

一眼题，LCT。

然后悲剧的发现不会写，只好分块来做。

令s = sqrt(n)，那么先分成s块，每块s个弹簧。

现在让每个点记录两个值，cnt和to，分别表示弹到这个块外面的次数和弹到了哪里。

我们发现单点修改只要修改块内元素，时间复杂度是O(s)的；而单点查询要查他后面的所有块，时间复杂度也是O(s)的，

于是总的复杂度是O(Q * logn)，解决！

/**************************************************************
    Problem: 2002
    User: rausen
    Language: C++
    Result: Accepted
    Time:1736 ms
    Memory:5700 kb
****************************************************************/
 
#include <cstdio>
#include <cmath>
#include <algorithm>
 
using namespace std;
 
int n, m, T, tot;
int w[250000], cnt[250000], to[250000], block[250000], a[250000];
 
void build(){
    int x = tot = 0, y;
    while (x < n){
        x += T, ++tot;
        w[tot] = x;
    }
    w[tot] = min(n, w[tot]);
     
    for (int i = tot; i >= 1; --i)
        for (int x = w[i]; x > w[i - 1]; --x){
            y = x + a[x];
            if (y <= w[i])
                to[x] = to[y], cnt[x] = cnt[y] + 1;
            else to[x] = y, cnt[x] = 1;
            block[x] = i;
        }
}
 
int query(int x){
    int res = 0;
    while (x <= n)
        res += cnt[x], x = to[x];
    return res;
}
 
void change(int x){
    int B = block[x], y;
    for (int i = x; i > w[B - 1]; --i){
        y = i + a[i];
        if (y <= w[B])
            to[i] = to[y], cnt[i] = cnt[y] + 1;
        else to[i] = y, cnt[i] = 1;
    }
}
 
int main(){
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i)
        scanf("%d", a + i);
    T = (int) sqrt(n);
    build();
 
    scanf("%d", &m);
    int X, Y, Z;
    while (m--){
        scanf("%d%d", &X, &Y);
        if (X == 1)
            printf("%d
", query(Y + 1));
        else {
            scanf("%d", &Z);
            a[Y + 1] = Z;
            change(Y + 1);
        }
    }
    return 0;
}

（p.s. 从0开始坑死人啊。。。。）