bzoj2002 [Hnoi2010]Bounce 弹飞绵羊【分块】

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=2002

这一题除了LCT解法，还有一种更巧妙，代码量更少的解法，就是分块。先想，如果仅仅记录每个节点需要几步可以弹飞，就可以做到O(1)查询O(n)修改；如果仅仅记录每个节点弹力洗漱，就可以做到O(n)查询O(1)修改。这会不会给人一种随机访问数组与链表的感觉呢？如果把n个弹簧分成√n块，记录块里每个弹簧需要几步才能跳出这一个块，并且记录跳出这个块后落到了哪，这样子查询以及修改复杂度都是O(√n）了。

#include <cstdio>
#include <cmath>

const int maxn = 200005;

int n, m, a[maxn], t1, t2, t3, siz, zuihou, kaishi, to[maxn], stp[maxn];

inline int qry(int pos) {
int rt = 0;
while (~pos) {
rt += stp[pos];
pos = to[pos];
}
return rt;
}
inline void upd(int pos, int data) {
kaishi = pos / siz * siz;
zuihou = (pos / siz + 1) * siz - 1;
a[pos] = data;
for (int i = pos; i >= kaishi; --i) {
if (i + a[i] >= n) {
to[i] = -1;
stp[i] = 1;
}
else if (i + a[i] > zuihou) {
to[i] = i + a[i];
stp[i] = 1;
}
else {
to[i] = to[i + a[i]];
stp[i] = stp[i + a[i]] + 1;
}
}
}

int main(void) {
//freopen("in.txt", "r", stdin);
scanf("%d", &n);
siz = (int)sqrt((float)n + 0.5f);
for (int i = 0; i < n; ++i) {
scanf("%d", a + i);
}
for (int i = n - 1; ~i; --i) {
zuihou = (i / siz + 1) * siz - 1;
if (i + a[i] >= n) {
to[i] = -1;
stp[i] = 1;
}
else if (i + a[i] > zuihou) {
to[i] = i + a[i];
stp[i] = 1;
}
else {
to[i] = to[i + a[i]];
stp[i] = stp[i + a[i]] + 1;
}
}
scanf("%d", &m);
while (m--) {
scanf("%d%d", &t1, &t2);
if (t1 == 1) {
printf("%d\n", qry(t2));
}
else {
scanf("%d", &t3);
upd(t2, t3);
}
}
return 0;
}