bzoj 2002 Bounce 弹飞绵羊

2002: [Hnoi2010]Bounce 弹飞绵羊

时间限制: 10 Sec  内存限制: 259 MB
提交: 16133  解决: 8253

题目描述

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

 

样例输入

4
1 2 1 1
3
1 1
2 1 1
1 1

样例输出

2
3

 

题解：分块。令k[i]表示为弹力系数，next[i]表示为跳出 i 所在块时之后的坐标，jump[i]表示为跳出 i 所在块需要的次数。由于绵羊是由 i 跳到 i+ki ,所以需从后往前处理数组。

 

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=200005;
int k[maxn],next[maxn],jump[maxn],block,n;
int main()
{
  int i,w,op,x,y,m,ans;
  scanf("%d",&n);
  block=sqrt(n);
  for(i=0;i<n;i++) scanf("%d",&k[i]);
  for(i=n-1;i>=0;i--)
   {
     w=i+k[i];
     if(w>=n) next[i]=-1,jump[i]=1;
     else if(w>=(i/block+1)*block) next[i]=w,jump[i]=1;
     else next[i]=next[w],jump[i]=jump[w]+1;
   }
  scanf("%d",&m);
  while(m--)
  {
    scanf("%d",&op);
    if(op==1)
    {
      ans=0;
      scanf("%d",&x);
      for(;next[x]!=-1;x=next[x]) ans+=jump[x];
      ans+=jump[x];
      printf("%d\n",ans);
    }
    else
    {
      scanf("%d",&x);scanf("%d",&k[x]);
      for(y=x;y>=x/block*block;y--)
      {
        w=y+k[y];
        if(w>=n) next[y]=-1,jump[y]=1;
        else if(w>=(y/block+1)*block) next[y]=w,jump[y]=1;
        else next[y]=next[w],jump[y]=jump[w]+1;
      }
    }
  }
  system("pause");
  return 0;
}