LOJ 6277. 数列分块入门 1 (分块)

题意

给出一个长为 nnn 的数列,以及 nnn 个操作,操作涉及区间加法,单点查值。

题解

分块,每次更新的话对块两侧不完整的地方暴力更新,对完整的块内部分打标记,查询时直接返回数据值和标记部分的值即可。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int nmax = 5e4 + 100 ;
const int INF = 0x3f3f3f3f;
const int mmax = sqrt(5e4 + 100) + 100;
ll data[nmax];
ll tag[mmax];
int n,block,num,L[mmax],R[mmax],belong[nmax];
void init(){
    block = sqrt(n);
    num = n / block; if(n % block) num++;
    for(int i = 1;i<=num;++i) L[i] = (i-1) * block +1, R[i] = i * block;
    for(int i = 1;i<=n;++i) belong[i] = (i-1) / block + 1;
}
int main() {
    scanf("%d",&n);
    init();
    for(int i = 1;i<=n;++i) scanf("%lld",&data[i]);
    for(int i = 1;i<=n;++i){
        int op,l,r; ll w;
        scanf("%d %d %d %lld",&op,&l,&r,&w);
        if(op == 0){
            if(belong[r] - belong[l] <= 1){
                for(int i = l;i<=r;++i) data[i] += w;
            }else{
                int lr = R[belong[l]], rl = L[belong[r]];
                for(int i = l;i<=lr;++i) data[i] += w;
                for(int i = rl;i<=r;++i) data[i] += w;
                for(int i = belong[l] + 1;i<= belong[r]-1;++i) tag[i] += w;
            }
        }else{
//            if(tag[belong[r]]){
//                ll ww = tag[belong[r]];
//                for(int i = L[belong[r]]; i<=R[belong[r]];++i) data[i] += ww;
//                tag[belong[r]] = 0;
//            }
            printf("%lld\n",data[r] + tag[belong[r]]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/pengwill97/article/details/80892533