bzoj2002

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec   Memory Limit: 259 MB
Submit: 4988   Solved: 2650
[ 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

分块，记录每个点第一次跳到下一块的步数和位置。每次按块跳，修改时只需要修改块内的那些位置。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#define Maxn 200010
using namespace std;

inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int l[Maxn],r[Maxn],belong[Maxn],a[Maxn];
int step[Maxn],pos[Maxn];
int cal(int x,int n){
    int ans=0;
    while(pos[x]<n){
        ans+=step[x];
        x=pos[x];
    }
    return ans+step[x];
}
int main()
{
    int n,m,op,x,y,sz,bk;
    n=read();
    for(int i=0;i<n;i++)
        a[i]=read();
    sz=(int)sqrt(n);
    if(n%sz) bk=n/sz+1;
    else bk=n/sz;
    for(int i=0;i<bk;i++)
        l[i]=i*sz,r[i]=l[i]+sz-1;
    r[bk-1]=n-1;
    for(int i=0;i<n;i++)
        belong[i]=i/sz;
    for(int i=n-1;i>=0;i--)
        if(i+a[i]>r[belong[i]]) step[i]=1,pos[i]=i+a[i];
        else step[i]=step[i+a[i]]+1,pos[i]=pos[i+a[i]];
    m=read();
    while(m--){
        op=read(),x=read();
        if(op==1)
            printf("%d\n",cal(x,n));
        else{
            y=read();
            a[x]=y;
            for(int i=x;i>=l[belong[x]];i--)
                if(i+a[i]>r[belong[i]]) step[i]=1,pos[i]=i+a[i];
                else step[i]=step[i+a[i]]+1,pos[i]=pos[i+a[i]];
        }
    }
	return 0;
}