【[HNOI2010]弹飞绵羊】

发现好像写了一个洛谷上最快的分块

这道题曾经一度感觉非常不可做,因为\(LCT\)的标签以及没有什么思路的分块

但是自从\(yy\)出来一个错误的哈希冲突分块之后(修改的时候挂掉了),就发现这道题不就是我曾经的那个错误的思路吗

这种要往后不断的跳的题目,我们暴力往后跳的话肯定是会爆炸的,因为这样的复杂度完全取决于询问

于是我们就分块好了,一次跳一个不行,那么我们就一次跳一个块好了

我们设\(b[i]\)表示从\(i\)这个位置开始跳,直到跳出所在块的跳跃次数是多少,\(c[i]\)表示\(i\)这个位置跳出所在块之后在哪一个位置

这个样子的话我们就能做到一次跳一个块了,于是现在询问的复杂度有了保证
这两个数组显然可以预处理出来,因为\(i\)往后跳一部肯定到达了\(i+a[i]\)这个位置,于是就有\(b[i]=b[i+a[i]]+1\)\(c[i]=c[i+a[i]]\),对于每一个块倒着预处理就好了

之后就是修改,因为这里只有单点修改,于是我们直接拎出那个块来,修改一下,像预处理那样暴力重构整个块就好了

复杂度也是\(O(\sqrt{n})\)

代码

#include
#include
#include
#include
#define re register
#define maxn 200005
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
    char c=getchar();
    int x=0;
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9')
        x=(x<<3)+(x<<1)+c-48,c=getchar();
    return x;
}
int a[maxn];
int b[maxn],c[maxn];
int l[maxn],r[maxn];
int len,Q,n,tot;
inline void Build_Block()
{
    len=std::sqrt(n);
    int k=1;
    while(k<=n)
    {
        l[++tot]=k;
        r[tot]=min(n,l[tot]+len-1);
        k=r[tot]+1;
        for(re int i=r[tot];i>=l[tot];--i)
            if(i+a[i]>r[tot]) b[i]=1,c[i]=i+a[i];
                else b[i]=b[i+a[i]]+1,c[i]=c[i+a[i]];
    }
}
inline int find(int x)
{
    if(x%len==0) return x/len;
    return x/len+1;
}
inline int query(int x)
{
    int ans=0;
    while(x<=n)
    {
        ans+=b[x];
        x=c[x];
    }
    return ans;
}
inline void change(int x,int val)
{
    int t=find(x);
    a[x]=val;
    for(re int i=x;i>=l[t];--i)
        if(i+a[i]>r[t]) b[i]=1,c[i]=i+a[i];
            else b[i]=b[i+a[i]]+1,c[i]=c[i+a[i]];
}
int main()
{
    n=read();
    for(re int i=1;i<=n;i++)
        a[i]=read();
    Build_Block();
    Q=read();
    int opt,x,y;
    while(Q--)
    {
        opt=read(),x=read();
        if(opt==1) printf("%d\n",query(x+1));
        else y=read(),change(x+1,y);
    }
    return 0;
}

你可能感兴趣的:(【[HNOI2010]弹飞绵羊】)