bzoj2002: [Hnoi2010]Bounce 弹飞绵羊

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

Sample Input

4

1 2 1 1

3

1 1

2 1 1

1 1

Sample Output

2

3

Solution

一道分块的好题目（不会lct的蒟蒻）。

具体怎么做呢？

先按照根号n为块大小进行分块

对于每一个点，计算出这个点跳出这个块要多少步，以及跳出这个块之后到了哪个点

那么我们就可以在根号n的时间内完成查询

对于修改，我们就把这个点相关的对应修改一下即可

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int sz=448,N=202000;
int x,k,i,j,n,m,t,ans;
int b
,a
,c
,be
;
int read(){
int sum=0;
char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9'){
sum=sum*10+c-'0';
c=getchar();}
return sum;
}
int main(){
//freopen("1.in","r",stdin);
//freopen("1.out","w",stdout);
n=read();
fo(i,1,n) a[i]=read(),be[i]=(i+sz-1)/sz;
b
=n+1;
fd(i,n,1)
if (be[i]==be[i+a[i]]) b[i]=b[i+a[i]],c[i]=c[i+a[i]]+1;
else c[i]=1,b[i]=i+a[i];
m=read();
while (m){
m--;
t=read();
if (t==1){
t=read(),t++,ans=0;
while (b[t]<=n) ans+=c[t],t=b[t];
ans+=c[t];
printf("%d\n",ans);
} else {
x=read(),k=read(),i=++x,a[x]=k,c[i]=0;
for(;x<=n&&be[x]==be[i];x+=a[x]) c[i]++; b[i]=x;
for(j=i-1;j>=1&&be[j]==be[i];j--)
if (be[j+a[j]]==be[i]) c[j]=c[j+a[j]]+1,b[j]=b[j+a[j]];
}
}
}