[HNOI2010] 弹飞绵羊

洛谷 P3203 传送门

昨天做分块（Ynoi）上瘾了，然后就看到了这道题......

LCT当然是正解，但是考场上如果能想出分块的做法当然是要写分块了啊。

毕竟我太弱了，LCT容易写挂。

分块多好写啊，只有50行.....

可以把这道题看作是经典数据结构题。

经典的数据结构题都有查询和修改两种操作。

对于这道题，如果直接暴力模拟，查询很慢，是O(n)的；但是修改很快，是O(1)的。

如果每次修改之后都预处理出所有点的答案，那么修改就会很慢，是O(n)；但是查询就很快了，达到O(1)。

在洛谷上看到了一句话：分块的作用就是平衡查询和修改的时间复杂度。

我们分块之后，可以处理出如果进到块里某个点，会跳几次才能出块，出块之后会跳到哪里。

这样对于每次询问，我们跳sqrt(n)次就行了。

对于每次修改，我们只需要重新处理被修改的点所在的块，也是O(sqrt n)。

这样查询和修改的复杂度就平衡了.....真和谐。

#include<cstdio>
#include<cmath>

int n,m,sz;
int bl[200005],l[635],r[635];
int k[200005],to[200005],cnt[200005];

void reset(int id)
{
    for(int i=r[id];i>=l[id];i--)
    {
        if(i+k[i]>r[id])to[i]=i+k[i],cnt[i]=1;
        else to[i]=to[i+k[i]],cnt[i]=cnt[i+k[i]]+1;
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)scanf("%d",&k[i]);
    sz=(int)sqrt(n*2);
    for(int i=0;i<n;i++)bl[i]=i/sz;
    for(int i=0;i<=bl[n-1];i++)
        l[i]=i*sz,r[i]=l[i]+sz-1;
    r[bl[n-1]]=n-1;
    for(int i=0;i<=bl[n-1];i++)reset(i);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int op,p,ck;
        scanf("%d%d",&op,&p);
        if(op==1)
        {
            int ans=0;
            while(p<n)
            {
                ans+=cnt[p];
                p=to[p];
            }
            printf("%d\n",ans);
        }
        if(op==2)
        {
            scanf("%d",&ck);
            k[p]=ck;
            reset(bl[p]);
        }
    }
    return 0;
}

 忘打 r[bl[n-1]]=n-1; 没一遍A......