BZOJ 2002 [Hnoi2010]Bounce 弹飞绵羊

题目描述#

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

输入#

第一行包含一个整数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

输出#

对于每个i=1的情况，你都要输出一个需要的步数，占一行。

样例输入#

4
1 2 1 1 3
1 1
2 1 1
1 1

样例输出#

2
3

题解#

将每一块都处理出
i 跳出这一块所用的次数和跳到的位置然后更新时将这一块重构

细节有问题的代码也能过（气）

1
#include <cstdio>
2
#include <cstring>
3
#include <algorithm>
4
#include <cmath>
5
inline int read()
6
{
7
    int x = 0, f = 1;
8
    char ch = getchar();
9
    while (ch > '9' || ch < '0')
10
    {
11
        if (ch == '-')
12
            f = -1;
13
        ch = getchar();
14
    }
15
    while (ch >= '0' && ch <= '9')
16
    {
17
        x = x * 10 + ch - '0';
18
        ch = getchar();
19
    }
20
    return x * f;
21
}
22
int n, m, k[200005], block;
23
int in[200005], next[200005], times[200005], l[200005];
24
int main()
25
{
26
    //freopen("1.in","r",stdin);
27
    //scanf("%d", &n);
28
    n = read();
29
    block = sqrt(n);
30
    for (int i = 0; i < n; i++)
31
    {
32
        k[i] = read();
33
        //scanf("%d", k + i);
34
        in[i] = i / block + 1;
35
        l[i] = (in[i] - 1) * block;
36
    }
37
    for (int i = n - 1; i >= 0; i--)
38
    {
39
        if (i + k[i] > n)
40
        {
41
            times[i] = 1;
42
            next[i] = n;
43
        }
44
        else if (in[i] == in[i + k[i]])
45
        {
46
            times[i] = times[i + k[i]] + 1;
47
            next[i] = next[i + k[i]];
48
        }
49
        else
50
        {
51
            times[i] = 1;
52
            next[i] = i + k[i];
53
        }
54
    }
55
    //scanf("%d", &m);
56
    m = read();
57
    int op, s;
58
    while (m--)
59
    {
60
        //scanf("%d", &op);
61
        op = read();
62
        if (op == 1)
63
        {
64
            //scanf("%d", &s);
65
            s = read();
66
            int ans = 0;
67
            for (int i = s; i < n; i = next[i])
68
            {
69
                ans += times[i];
70
            }
71
            printf("%d\n", ans);
72
        }
73
        else
74
        {
75
            //scanf("%d%d", &s, &c);
76
            //k[s] = c;
77
            s = read();
78
            k[s] = read();
79
            for (int i = s; i >= l[s]; i--)
80
            {
81
                if (i + k[i] > n)
82
                {
83
                    times[i] = 1;
84
                    next[i] = i + k[i];
85
                }
86
                if (in[i] == in[i + k[i]])
87
                {
88
                    times[i] = times[i + k[i]] + 1;
89
                    next[i] = next[i + k[i]];
90
                }
91
                else
92
                {
93
                    times[i] = 1;
94
                    next[i] = i + k[i];
95
                }
96
            }
97
        }
98
    }
99
}