loj#6278. 数列分块入门 2

#6278. 数列分块入门 2

题目：传送门 

简要题意：

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

 

---

 

 

题解：

　　 分块之后直接就分情况暴力了啊：

　　 先枚举头，再枚举尾，中间的块之间二分求位置（当然了，预处理就要排序一下）

 

---

 

 

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
int n,a[51000],ba[51000];
int block,pos[51000];
vector<int> v[510];
void reset(int x)
{
    v[x].clear();
    for(int i=(x-1)*block+1;i<=x*block;i++)
        v[x].push_back(a[i]);
    sort(v[x].begin(),v[x].end());
}
void update(int l,int r,int c)
{
    for(int i=l;i<=min(pos[l]*block,r);i++)a[i]+=c;
    reset(pos[l]);
    if(pos[l]!=pos[r])
        for(int i=(pos[r]-1)*block+1;i<=r;i++)
            a[i]+=c;
    reset(pos[r]);
    for(int i=pos[l]+1;i<=pos[r]-1;i++)ba[i]+=c;
}
void sol(int l,int r,int c)
{
    int ans=0;
    for(int i=l;i<=min(pos[l]*block,r);i++)if(a[i]+ba[pos[i]]<c)ans++;
    if(pos[l]!=pos[r])
        for(int i=(pos[r]-1)*block+1;i<=r;i++)
            if(a[i]+ba[pos[i]]<c)
                ans++;
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
    {
        int x=c-ba[i];
        ans+=lower_bound(v[i].begin(),v[i].end(),x)-v[i].begin();
    }
    printf("%d\n",ans);
}
int main()
{
    memset(pos,0,sizeof(pos));memset(ba,0,sizeof(ba));
    scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    block=int(sqrt(n));for(int i=1;i<=n;i++){pos[i]=(i-1)/block+1;v[pos[i]].push_back(a[i]);}
    for(int i=1;i<=pos[n];i++)sort(v[i].begin(),v[i].end());
    for(int i=1;i<=n;i++)
    {
        int opt,l,r,c;
        scanf("%d%d%d%d",&opt,&l,&r,&c);
        if(opt==0)update(l,r,c);
        else sol(l,r,c*c);
    }
    return 0;
}