原题链接
Bounce 弹飞绵羊
Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 8285 Solved: 4266
[Submit][Status][Discuss]
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
这个题其实是一个标准的LCT或者是伸展树的问题,但是用分块来解会更加方便。关于块状链表,就是把一个长为n数组分为√n份,把每一份作为一个相对独立的单元记算。也就是分治法的思想。对于此题,我们用s[]表示距离;belong[]表示此点属于哪一块;nex[]表示此点的后继;k[]表示此点的劲度系数。对于同一连通块内的两点,如果前一点可以跳到后一点,那么前一点的w[]应等于后一点的w[],而前一点的s[]应为后一点的s[]+1。如果是不同连通块里的两点,前一点可以到达后一点,那么前一点的w[]指向后一点,前一点的s[]值为1。求值时累加。为什么分块后会更快,分块之后,通过把块内能到达的部分都进行了整合,同时指向了一个块内的祖先,这样原本打算指向块内的第一步,现在就会直接走动这一块的祖先上去,中间节省了很多步,所以得到了优化。当然跨块的时候那一步就没有办法避免了。但是如果不分块,一个位置可能就是由无数个位置走过来的,但是中间的重复计算却得不到有效的简化
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200000 + 10;
int n,m,cmd;
int k[maxn],nex[maxn],s[maxn],belong[maxn];
//k代表弹性系数,next代表该元素的后继,s表示中间的距离,belong的值表示属于哪一个块
void make(int i){
int t=i+k[i];
if(t>=n) t=n;//到了n就是边界了
if(belong[i]==belong[t]){
nex[i]=nex[t];//相同块内的后继都是一样的
s[i]=s[t]+1;//这个距离是倒过来算的,从后面往前面算距离
}
else{
nex[i]=t;//跨块的时候就是直接将后继指向下一个块
s[i]=1;//跨块的时候的距离就是1了,这个距离后面总要加上
}
}
int main(){
while(scanf("%d",&n)==1){
for(int i=0;i<n;i++)
scanf("%d",&k[i]);
int mod=(int)sqrt(n);
for(int i=(n-1);i>=0;i--)
belong[i]= (i/mod);//将每个元素所在的块进行分类
for(int i=(n-1);i>=0;i--)
make(i);//对该元素进行分块处理
scanf("%d",&m);
while(m--){
scanf("%d",&cmd);
if(cmd==1){
int b,ans=0;
scanf("%d",&b);
for(int i=b;i<n;i=nex[i])
ans+=s[i];//只需要不断地将这些距离进行相加即可
printf("%d\n",ans);
}
else{
int loc,t;
scanf("%d%d",&loc,&t);
k[loc]=t;
for(int i=loc;i>=0;i--){
if(belong[i]==belong[loc]) make(i);
else break;
}
}
}
}
return 0;
}