BZOJ 2002 分块

简略题意：某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

假若暴力跳转统计，每次询问复杂度是$O(n)$的，每次修改复杂度是$O(1)$的。
假若DP预处理询问，每次询问复杂度是$O(1)$的，每次修改复杂度是$O(n)$的。
因此考虑折中分块，每块的询问单独预处理，询问时跳转每个块，复杂度$O(\sqrt n)$，每次修改需要变动的只有块中元素，重新处理这个块内的DP值，复杂度$O(\sqrt n)$。
因此总复杂度$O(m\sqrt(n))$。

#include <bits/stdc++.h>

using namespace std;

const int maxn = 220000;

int n, m;
int block;
int L[500], R[500], belong[maxn];
int to[maxn], dp[maxn];
int skip[maxn];

int solve(int pos) {
    int res = 0;
    while(pos < n) {
        res += dp[pos];
        pos = to[pos];
    }
    return res;
}

int main() {
    while(~scanf("%d", &n)) {
        memset(to, -1, sizeof to);
        memset(dp, 0, sizeof dp);
        block = sqrt(n) + 1;

        for(int i = 0; i < n; i++) {
            scanf("%d", &skip[i]);
            belong[i] = i / block;
        }
        for(int i = 0; i <= block; i++) {
            L[i] = i*block;
            R[i] = L[i] + block - 1;
        }
        for(int i = n - 1; i >= 0; i--) {
            if(belong[i] == belong[i+skip[i]]) {
                to[i] = to[i+skip[i]];
                dp[i] = dp[i+skip[i]] + 1;
            } else {
                to[i] = i+skip[i];
                dp[i] = 1;
            }
        }
        scanf("%d", &m);
        for(int i = 1; i <= m; i++) {
            int c;
            scanf("%d", &c);
            if(c == 1) {
                int v;
                scanf("%d", &v);
                printf("%d\n", solve(v));
            } else {
                int u, v;
                scanf("%d%d", &u, &v);
                skip[u] = v;
                for(int i = u; i >= L[belong[u]]; i--) {
                    if(belong[i] == belong[i+skip[i]]) {
                        dp[i] = dp[i+skip[i]] + 1;
                        to[i] = to[i+skip[i]];
                    } else {
                        dp[i] = 1;
                        to[i] = i + skip[i];
                    }
                }
            }
        }
    }
    return 0;
}