BZOJ 2002 Bounce 弹飞绵羊（分块）

最新推荐文章于 2020-03-26 00:42:10 发布

原创最新推荐文章于 2020-03-26 00:42:10 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

分块专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一种使用分块思想解决特定查询更新问题的方法。通过预先计算每个块的结束位置和剩余步数，可以高效地处理数组上的查询与更新操作。适用于需要频繁进行范围查询和更新的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：点击打开链接

思路：

利用分块思想。

划分块左右界的方式是固定的，加俩数组即可。

en[i]表示跳出i所在块后的位置，ne[i]表示跳出i所在的块还有几步。

#include<bits/stdc++.h>
using namespace std;

const int maxn=200005;
int a[maxn],en[maxn],ne[maxn],l,r,n,m,block,p;

int query(int pos){
    int ans=0;
    while(pos<=n){
        ans+=ne[pos];
        pos=en[pos];
    }
    return ans;
}
void update(int pos,int x){
    a[pos]=x;
    l=(pos/p)*p+1; //找出该块最左边的下标
    r=l+p-1;
    if(pos%p==0) l-=p,r-=p; //如果是块的最后一个如此处理
    for(int i=pos;i>=l;i--){
        int u=i+a[i];
        if(u>r){
            en[i]=u;
            ne[i]=1;
        }
        else{
            en[i]=en[u];
            ne[i]=ne[u]+1;
        }
    }
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    block=sqrt(n),p=n/block; //有几块，一块有几个
    if(n%p) block++;

    for(int i=1;i<=block;i++){
        l=(i-1)*p+1;
        r=min(i*p,n);
        for(int j=r;j>=l;j--){  //倒着找，可以使已经找到的赋给前面
            int u=j+a[j];
            if(u>r){
                en[j]=u;  //表示跳出i所在块的位置
                ne[j]=1;  //表示距离跳出i所在的块还有几步
            }
            else{
                en[j]=en[u];
                ne[j]=ne[u]+1;
            }
        }
    }
    scanf("%d",&m);
    int flg,pos,x;
    for(int i=1;i<=m;i++){
        scanf("%d",&flg);
        if(flg==1){
            scanf("%d",&pos);
            printf("%d\n",query(pos+1));
        }
        else{
            scanf("%d%d",&pos,&x);
            update(pos+1,x);
        }
    }
    return 0;
}