LOJ #6278. 数列分块入门 2

hzwer tql!

渐渐找到了分块的套路。

给出一个长为  n 的数列，以及 n 个操作，操作涉及区间加法，询问区间内小于某个值 c*c 的元素个数。

残块先在原数组上暴力，然后拿原数组更新块数组（效率高于结构体存id！），整块就标记。

code：

#include<iostream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<vector>
#define pb push_back
#define lb lower_bound
using namespace std;

const int Maxn = 50005; 

vector<int> V[Maxn];
int bl[Maxn],v[Maxn],atag[Maxn];
int n,blo,l,r,opt,c;

void reset(int x){
    for(int i = (x-1)*blo+1;i <= min(n,x*blo);i++)
        V[x][(i-1)%blo] = v[i];
    sort(V[x].begin(),V[x].end());
} 

void add(int l,int r,int c){
    if(bl[l] == bl[r]){
        for(int i = l;i <= r;i++)v[i] += c;
        reset(bl[r]);return;
    }
    for(int i = l;i <= min(n,bl[l]*blo);i++)v[i] += c;reset(bl[l]);
    for(int i = (bl[r]-1)*blo+1;i <= r;i++)v[i] += c;reset(bl[r]);
    for(int i = bl[l]+1;i < bl[r];i++)atag[i] += c;
}

int ask(int l,int r,int c2){
    if(l > r)return 0;
    int cnt = 0;
    if(bl[l] == bl[r]){
        for(int i = l;i <= r;i++)if(v[i]+atag[bl[i]] < c2)cnt++;
        return cnt;
    }
    for(int i = l;i <= min(bl[l]*blo,n);i++)
        if(v[i]+atag[bl[i]] < c2)cnt++;
    for(int i = (bl[r]-1)*blo+1;i <= min(r,n);i++)
        if(v[i]+atag[bl[i]] < c2)cnt++;
    for(int i = bl[l]+1;i < bl[r];i++)
        cnt += lb(V[i].begin(),V[i].end(),c2-atag[i])-V[i].begin();
    return cnt;
}

int main(){
    scanf("%d",&n);blo = sqrt(n);
    for(int i = 1;i <= n;i++){
        scanf("%d",&v[i]);
        bl[i] = (i-1)/blo+1;
        V[bl[i]].pb(v[i]);
    }
    for(int i = 1;i <= bl[n];i++)sort(V[i].begin(),V[i].end());
    for(int i = 1;i <= n;i++){
        scanf("%d%d%d%d",&opt,&l,&r,&c);
        if(opt)printf("%d
",ask(l,r,c*c));
        else add(l,r,c);
    }
return 0;
}