数列分块入门 2（LibreOj-6278）

【题目描述】

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

【输入格式】

第一行输入一个数字 n。

第二行输入 n 个数字，第 i 个数字为 ai，以空格隔开。

接下来输入 n 行询问，每行输入四个数字 opt、l、r、c，以空格隔开。

若 opt=0，表示将位于 [l,r] 的之间的数字都加 c。

若 opt=1，表示询问位于 [l,r] 中小于 c^2 的数字的个数

【输出格式】

对于每次询问，输出一行一个数字表示答案。

【样例】

样例输入
4
1 2 2 3
0 1 3 1
1 1 3 2
1 1 4 1
1 2 3 2

样例输出
3
0
2

【数据范围与提示】

对于 100% 的数据，1<=n<=50000,-2^31<=other,ans<=2^31-1。

【源代码】

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 1E9+7;
const int N = 100000+5;
const int dx[] = {-1,1,0,0,-1,-1,1,1};
const int dy[] = {0,0,-1,1,-1,1,-1,1};
using namespace std;
int n;
int block,sum;//block为块的长度,sum为块的个数
int a[N];//存放数列元素
int pos[N],tag[N];//pos记录第i个元素在第几个块中,tag为操作标记
int ans[N];//维护整块和
vector<int> v[N];
void init(){
    block=sqrt(n);//块的长度
    sum=n/block;//块个数
    if(n%block)
        sum++;
    for(int i=1;i<=n;i++){
        pos[i]=(i-1)/block+1;//第i个元素在第几块中
        v[pos[i]].push_back(a[i]);//保存每个数分块的序号
    }

    for(int i=1;i<=sum;i++)//对整块进行排序
        sort(v[i].begin(),v[i].end());
}
void resort(int x){
    v[pos[x]].clear();
    for(int i=(pos[x]-1)*block+1;i<=min(pos[x]*block,n);i++)
        v[pos[x]].push_back(a[i]);
    sort(v[pos[x]].begin(),v[pos[x]].end());
}
void update(int L,int R,int x){
    for(int i=L;i<=min(pos[L]*block,R);i++)//左边的边角料
        a[i]+=x;
    resort(L);//对不完整块排序
    if(pos[L]!=pos[R]){//存在右区间才遍历，防止重复计算
        for(int i=(pos[R]-1)*block+1;i<=R;i++)//右边的边角料
            a[i]+=x;
        resort(R);//对不完整块排序
    }
    for(int i=pos[L]+1;i<=pos[R]-1;i++)//中间的整块
        tag[i]+=x;
}
int query(int L,int R,int x){
    int res=0;

    for(int i=L;i<=min(pos[L]*block,R);i++)//左边的边角料
        if(a[i]+tag[pos[i]]<x)
            res++;

    if(pos[L]!=pos[R])//存在右区间才遍历，防止重复计算
        for(int i=(pos[R]-1)*block+1;i<=R;i++)//右边的边角料
            if(a[i]+tag[pos[i]]<x)
                res++;

    for(int i=pos[L]+1;i<=pos[R]-1;i++){//中间的整块进行二分查找
        int temp=x-tag[i];
        res+=lower_bound(v[i].begin(),v[i].end(),temp)-v[i].begin();
    }
    return res;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);

    init();

    for(int i=1;i<=n;i++){
        int op;
        int left,right,x;
        scanf("%d",&op);
        scanf("%d%d%d",&left,&right,&x);
        if(op==0)
            update(left,right,x);
        else if(op==1)
            printf("%d\n",query(left,right,x*x));
    }
    return 0;
}