BZOJ2002 [Hnoi2010] Bounce 弹飞绵羊

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

这道题有两个坑点

1.弹簧是0--n-1，而我们习惯是从1--n.

2.更新时要从当前块的末尾向前更新就可以。如果从前往后会使用错误的弹力值更新，我这样做在洛谷上只拿了30分

P3203 [HNOI2010]弹飞绵羊 

分析下思路：

因为不会LCT，所以用分块暴力。

f[i]表示在当前点i需要跳多少次才可以跳出当前块。

to[i]表示从i点开始跳，跳出当前块达到所在点的编号。

从后向前更新就好了。

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 200001
int belong[N];
int k[N];
int f[N];
int to[N];
int n,Q,q,K;
int block,x;
void update(int here,int delta)
{
    int block=sqrt(n);
    k[here]=delta;
    for(int i=min(belong[here]*block,n);i>=(belong[here]-1)*block+1;i--)
    {
        to[i]=belong[i+k[i]] == belong[i] ? to[i+k[i]] : i+k[i];
        f[i]=belong[i+k[i]] == belong[i] ? (f[i+k[i]]+1) : 1;
    }
}
int query(int x)
{
    int ans=0;
    while(x<=n)
    {
        ans+=f[x];
        x=to[x];
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    int block=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&k[i]);
        belong[i]=(i-1)/block+1;
    }
    for(int i=n;i>0;i--)
    {
        to[i]=belong[i+k[i]] == belong[i] ? to[i+k[i]] : i+k[i];
        f[i]=belong[i+k[i]] == belong[i] ? (f[i+k[i]]+1) : 1;        
    }
    scanf("%d",&Q);
    for(int i=1;i<=Q;i++)
    {
        scanf("%d",&q);
        if(q==1)
        {
            scanf("%d",&x);
            x++;
            printf("%d\n",query(x));
        }
        else
        {
            scanf("%d%d",&x,&K);
            x++;
            update(x,K);
        }
    }
}