数列分块入门2

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

/*对于在一个区间内小于c的元素的个数问题，用分块方法求解，
  我们可以进行这样的预处理：把每个块的元素进行排序：
  这样找的话，对于两端没有在块里的元素暴力计算，块内的只需要二分，
  可以引用vector，记录每个块排序过后的序列
*/
#include<stdio.h>
#include<string.h>
#include<string>
#include<iostream>
#include<algorithm>
#include<queue>
#include<math.h>
#include<map>
#include<vector>
#include<stack>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N=50005;

int a[N],be[N],add[N];
int n,mod;
vector<int>v[N];//记录每个块中的数字排序过后的排列

void reset(int x)//每次进行加法操作，对于两边不在块中
{                //的数字，要将它所在的块中元素再一次排序
    v[x].clear();
    for(int i=(x-1)*mod+1; i<=min(x*mod,n); i++)
        v[x].push_back(a[i]);
    sort(v[x].begin(),v[x].end());
}

void ADD(int l,int r,int k)
{
    int i;
    for(i=l; i<=min(r,mod*be[l]); i++)
        a[i]+=k;
    reset(be[l]);//对不在块中的元素的所在块再次排序
    if(be[r]!=be[l])
    {
        for(i=(be[r]-1)*mod+1; i<=r; i++)
            a[i]+=k;
        reset(be[r]);//同上
    }
    for(i=be[l]+1; i<=be[r]-1; i++)
        add[i]+=k;
}

int qurey(int l,int r,int c)
{
    int i,ans=0;
    for(i=l; i<=min(r,mod*be[l]); i++)
        if(a[i]+add[be[i]]<c) ans++;
    if(be[l]!=be[r])
        for(i=(be[r]-1)*mod+1; i<=r; i++)
            if(a[i]+add[be[i]]<c) ans++;
    for(i=be[l]+1; i<=be[r]-1; i++)
    {
        int temp=c-add[i];
        ans+=lower_bound(v[i].begin(),v[i].end(),temp)-v[i].begin();
    }
    return ans;
}

int main()
{
    int i,temp,l,r,k,op;
    scanf("%d",&n);
    mod=sqrt(n);
    for(i=1; i<=n; i++)
        scanf("%d",&a[i]);
    for(i=1; i<=n; i++)
        be[i]=(i-1)/mod+1;
    for(i=1; i<=n; i++)
        v[be[i]].push_back(a[i]);
    for(i=1; i<=be[n]; i++)
        sort(v[i].begin(),v[i].end());
    for(i=0; i<n; i++)
    {
        scanf("%d %d %d %d",&op,&r,&l,&k);
        if(op==0)
            ADD(r,l,k);
        else
            printf("%d\n",qurey(r,l,k*k));
    }
    return 0;
}