bzoj 2002 弹飞绵羊 分块 解题报告

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

思路

分块，pt【跳一次的位置】，st【跳出块需要多少步】。然后看看是不是出块了，分情况更新值。具体看代码吧，显而易见的。。。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=200000+5;
int n,m,num,now,lf[N],rt[N],belong[N],st[N],pt[N],line[N];
int calcu(int x)
{
    int t=0;
    while(true)
    {
        t+=st[x];
        if (pt[x]==0) break;
        x=pt[x];
    }
    return t;
}
int main()
{
    scanf("%d",&n);
    now=sqrt(n);
    for (int i=1;i<=n;i++)
    scanf("%d",&line[i]);
    if (n%now==1) num=n/now+1;
    else num=n/now;
    for (int i=1;i<=num;i++)
    {
        lf[i]=(i-1)*now+1;
        rt[i]=i*now;
    }
    rt[num]=n;
    for (int i=1;i<=n;i++)
    belong[i]=(i-1)/now+1;
    for (int i=n;i>0;i--)
    {
        if (i+line[i]>n) st[i]=1;
        else if (belong[i]==belong[i+line[i]])
        {
            st[i]=st[i+line[i]]+1;
            pt[i]=pt[i+line[i]];
        }
        else 
        {
            st[i]=1;
            pt[i]=i+line[i];
        }
    }
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        int x,y,f;
        scanf("%d%d",&f,&x);
        x++;
        if (f==1) printf("%d\n",calcu(x));
        else 
        {
            scanf("%d",&y);
            line[x]=y;
            for (int i=x;i>=lf[belong[x]];i--)
            if (belong[i]==belong[i+line[i]])
            {
                st[i]=st[i+line[i]]+1;
                pt[i]=pt[i+line[i]];
            }
            else 
            {
                st[i]=1;
                pt[i]=i+line[i];
            }
        }
    }
    return 0;
}