LOJ #6278. 数列分块入门 2 (分块/树状数组

题意:

给出一个长为 $n$的数列，以及 $n$ 个操作，操作涉及区间加法，询问区间内小于某个值 x x <script type="math/tex" id="MathJax-Element-86">x</script> 的元素个数。

分块, 直接维护每个块二分即可,
树状数组, 代码写起来比分块还简单,,,,

分块

//
// Created by team02 on 18-8-17.
//
#include <bits/stdc++.h>

using namespace std;

#define ll long long
#define pb push_back
#define ls o<<1
#define rs o<<1|1
#define fi first
#define se second
#define CLR(a, b) memset(a, (b), sizeof(a))
const int INF = 0x3f3f3f3f;
const int mod = 1e9+7;
const int MAXN = (int)5e4+10;

int a[MAXN],b[MAXN],c[MAXN];
vector<int> v[500];
int n,k;
inline void update(int l,int r,int x){
    for(int i=l;i<=min(b[l]*k,r);++i) a[i]+=x;
    v[b[l]].clear();
    for(int i=(b[l]-1)*k+1;i<=min(b[l]*k,n);++i) v[b[l]].pb(a[i]);
    sort(v[b[l]].begin(),v[b[l]].end());
    if(b[l]!=b[r]) {
        for(int i=(b[r]-1)*k+1;i<=r;++i) a[i]+=x;
        v[b[r]].clear();
        for(int i=(b[r]-1)*k+1;i<=min(b[r]*k,n);++i) v[b[r]].pb(a[i]);
        sort(v[b[r]].begin(),v[b[r]].end());
    }
    for(int i=b[l]+1;i<=b[r]-1;++i) c[i]+=x;
}
inline int query(int l,int r,int x) {
    int kk=0;
    for(int i=l;i<=min(b[l]*k,r);++i)
        if(a[i]+c[b[l]]<x) ++kk;
    if(b[l]!=b[r])
        for(int i=(b[r]-1)*k+1;i<=r;++i)
            if(a[i]+c[b[r]]<x) ++kk;
    for(int i=b[l]+1;i<=b[r]-1;++i) {
        int xx=x-c[i];
        kk+=(lower_bound(v[i].begin(),v[i].end(),xx)-v[i].begin());
    }
    return kk;
}

int main() {
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;++i) cin>>a[i];
    k=(int)sqrt(n);
    for(int i=1;i<=n;++i) b[i]=(i-1)/k+1, v[b[i]].pb(a[i]);
    for(int i=1;i<=b[n];++i) sort(v[i].begin(),v[i].end());
    for(int i=1;i<=n;++i){
        int op,l,r,x;
        cin>>op>>l>>r>>x;
        if(op==0) update(l,r,x);
        else cout<<query(l,r,x*x)<<endl;
    }
    return 0;
}

树状数组