【bzoj2002】[Hnoi2010]Bounce 弹飞绵羊

bzoj2002 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.动态树不太会》》》分块乱搞（10s）；
2.以sqrt（n）的大小为块，记录它跳出此块的步数，与跳出到的点————方便sqrt（n） 的while查询；
3.在线修改可只修改i块（与i相关的）点 的*pos，*step；

#include<iostream>
#include<stdio.h>
#include<cmath>
using namespace std;
int k[200005];
int n,m;
int be[200005];
int bo;
struct node{
    int step,pos;
} q[200005];
int solve(int x)
{
    int ans=0;
    while(x<=n)
    {   
        ans+=q[x].step;
        x=q[x].pos;
    }
    return ans;
}
int change(int x,int y)
{

    k[x]=y;
    int j=x+k[x];
    if(be[x+y]!=be[x]) 
    {
        q[x].pos=x+y,q[x].step=1;
    }
    else q[x].pos=q[x+y].pos,q[x].step=q[x+y].step+1;

    int now=x-1;
    int noo=be[x];
    while(noo==be[now])
    {
        if(be[now+k[now]]==noo)
        {
            q[now].step=q[now+k[now]].step+1;
            q[now].pos=q[now+k[now]].pos;
        } 
        now--;
    }
}
int main()
{
    scanf("%d",&n);
    bo=sqrt(n);
    for(int i=1;i<=n;i++)  
    {
        scanf("%d",&k[i]);
        be[i]=(i-1)/bo+1;
    }
    for(int i=n;i>=1;i--)
    {
        int j=i+k[i];
        if(j>n||be[j]>be[i]) q[i].step=1,q[i].pos=j;
        else q[i].step=q[j].step+1,q[i].pos=q[j].pos;
    }
    scanf("%d",&m);
    int x,y,z;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        y++;
        if(x==1) printf("%d\n",solve(y));
        if(x==2)
        {
            scanf("%d",&z);
            change(y,z);
        }   
    }
}