Bounce 弹飞绵羊 HYSBZ - 2002 【分块】

传送门

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的情况，你都要输出一个需要的步数，占一行。

思路很巧妙，题目上要求跳到 > n 的位置所需要的次数，还涉及到更新的问题，通过分块我们将给的 n 的分为 sqrt(n) 的块，将跳转为 > n 的问题转化为 跳出自己所在块所需要的步数。

我们需要记录两个东西，跳出自己这个块所需要的步数和跳出去后所在的位置，我们通过 O(n) 的复杂度处理出原本数组的信息。
如果需要求解一个位置跳到 >n 的位置，我们直接一个 while 循环即可求出答案.
求解答案的复杂度 O( sqrt(n) )。
如果要更新一个点的弹力系数，我们只需要更新这个块里面的装置（在这个点之前的装置）。
更新的复杂度也是 O( sqrt(n) )。

///#include <bits/stdc++.h>
///#include <unordered_map>
///#include <unordered_set>
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <set>
#include <stack>
#include <map>
#include <new>
#include <ctime>
#include <vector>

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai = acos(-1.0);
const double E = exp(1.0);
const int INF = 0x3f3f3f3f;
const ll mod = 10007;

int n, q, cnt;
int pos[200500];
struct node {
    int k;  ///弹力系数
    int step;   ///跳出该块的步数
    int sub;    ///跳出后所在的下标
} bounce[200500];

void block() {
    cnt = sqrt(n);
    for (int i = 1; i <= n; i++) {
        pos[i] = (i + cnt - 1) / cnt;
    }
}

int main() {
    int op, l, x;
    while (scanf("%d", &n) != EOF) {
        for (int i = 1; i <= n; i++) {
            scanf("%d", &bounce[i].k);
        }
        block();
        int s, sub;
        for (int i = n; i >= 1; i--) {
            if (pos[bounce[i].k + i] != pos[i]) {
                bounce[i].step = 1;
                bounce[i].sub = i + bounce[i].k;
            } else {
                bounce[i].step = bounce[i + bounce[i].k].step + 1;
                bounce[i].sub = bounce[i + bounce[i].k].sub;
            }
        }
        scanf("%d", &q);
        while (q--) {
            scanf("%d", &op);
            if (op == 1) {
                scanf("%d", &l);
                l++;
                int ans = 0;
                while (l <= n) {
                    ans += bounce[l].step;
                    l = bounce[l].sub;
                }
                printf("%d\n", ans);
            } else {
                scanf("%d %d", &l, &x);
                l++;
                bounce[l].k = x;
                for (int i = l; i >= (pos[l] - 1) * cnt + 1; i--) {
                    if (pos[bounce[i].k + i] != pos[i]) {
                        bounce[i].step = 1;
                        bounce[i].sub = i + bounce[i].k;
                    } else {
                        bounce[i].step = bounce[i + bounce[i].k].step + 1;
                        bounce[i].sub = bounce[i + bounce[i].k].sub;
                    }
                }
            }
        }
    }
    return 0;
}