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

输入输出样例

输入样例#1：
4
1 2 1 1
3
1 1
2 1 1
1 1
输出样例#1：
2
3

蒟蒻不会LCT，只能用分块写了 QAQ

program df;
var i,j,n,m,x,y,z,k,t,xx,l,r:longint;
a,b,c,d:array[0..200000] of longint;

procedure check1(x:longint);
var i:longint;
begin
i:=0;
while x<=n do
begin
i:=i+d[x];
x:=c[x];
end;
writeln(i);
end;

procedure check(x,y:longint);
var i,j,l,r,xx:longint;
begin
a[x]:=y;
l:=(x div k)*k+1;
r:=l+k-1; //找到x所在区间
if x mod k=0 then //此时x在上一个区间的末尾，所以属于上个区间（区间为左闭右开）
begin
l:=l-k;
r:=r-k;
end;
for i:=x downto l do //更新跳出这个块的时间
begin
xx:=i+a[i];
if xx>r then
begin
c[i]:=xx;
d[i]:=1;
end
else
begin
c[i]:=c[xx];
d[i]:=d[xx]+1;
end;
end;
end;

begin
readln(n);
for i:=1 to n do
read(a[i]);
t:=trunc(sqrt(n));
k:=n div t;
if n mod k<>0 then inc(t);
for i:=1 to t do
begin
l:=(i-1)*k+1;
r:=i*k;
if r>n then r:=n;
for j:=r downto l do
begin
xx:=j+a[j];
if xx>r then
begin
c[j]:=xx; //j在下一个块的位置
d[j]:=1; //跳出这个块的时间为1
end
else
begin
c[j]:=c[xx]; //j在下一个块的位置与xx在下个块的位置相同
d[j]:=d[xx]+1; //j跳出的时间是xx跳出的时间+1
end;
end;
end;
readln(m);
for i:=1 to m do
begin
read(x);
if x=1 then begin readln(y); check1(y+1); end;
if x=2 then
begin
readln(y,z);
check(y+1,z);
end;
end;
end.