COGS.1689.[HNOI2010]Bounce 弹飞绵羊(分块)

题目链接

/*
分块，维护从某位置跳到下一个块需要多少步 
*/
#include<cmath>
#include<cstdio>
#include<cctype>
using namespace std;
const int N=2e6+5;

int n,v[N],belong[N],size,tot,step[N]/*跳出这个块需要多少步*/,nxt[N]/*最后跳到这块的哪*/;

inline int read()
{
    int now=0,f=1;register char c=getchar();
    for(;!isdigit(c);c=getchar())
      if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=getchar());
    return now*f;
}

void Modify(int p)
{
    int val=read();
    v[p]=val;
//不能只改一个点！因为同一块中，前面是由后面得到的，且维护的nxt[aft]，即aft跳到下一个块的位置，所以必须从pos往前再更新一遍 
    for(int i=p;i>(belong[p]-1)*size;--i)
    {
        int aft=i+v[i];
        if(aft>n)//可能会在边界 
            step[i]=1, nxt[i]=n+1;
        else if(belong[aft]!=belong[i])
            step[i]=1, nxt[i]=aft;
        else
            step[i]=step[aft]+1, nxt[i]=nxt[aft];
    }
//  printf("Modify  %d:v:%d aft:%d step:%d nxt:%d\n",p,v[p],aft,step[p],nxt[p]);
}
int Query(int p)
{
    int res=0;
    while(p<=n)
//      printf("pos:%d step:%d nxt:%d\n",p,step[p],nxt[p]),
        res+=step[p], p=nxt[p];
    return res;
}

int main()
{
//  freopen("bzoj_2002.in","r",stdin);
//  freopen("bzoj_2002.out","w",stdout);

    n=read();size=sqrt(n);
//  tot=size;//n/size = n/sqrt(n) = sqrt(n) = size
//  if(n%size) ++tot;
    for(int i=1;i<=n;++i)
        v[i]=read(), belong[i]=(i-1)/size+1;
    for(int i=n;i;--i)//逆序做，这样就能利用前边得到的值O(1)得到某结果 
    {
        int aft=i+v[i];
        if(aft>n)
            step[i]=1, nxt[i]=n+1;
        else if(belong[aft]!=belong[i])
            step[i]=1, nxt[i]=aft;
        else
            step[i]=step[aft]+1, nxt[i]=nxt[aft];
//      printf("%d:v:%d aft:%d step:%d nxt:%d\n",i,v[i],aft,step[i],nxt[i]);
    }
    int m=read(),opt,p;
    while(m--)
    {
        opt=read(),p=read()+1;
        if(opt==1)
            printf("%d\n",Query(p));
        else
            Modify(p);
    }

//  fclose(stdin);fclose(stdout);
    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/8435089.html