题目链接:https://loj.ac/problem/6278
题目大意:
询问操作: 不完整的块枚举统计即可;而要在每个整块内寻找小于一个值的元素数,于是我们不得不要求块内元素是有序的,这样就能使用二分法对块内查询,需要预处理时每块做一遍排序,复杂度O(nlogn),每次查询在√n个块内二分,以及暴力2√n个元素,总复杂度O(nlogn + n√nlog√n)。
每次修改,边块的vector必须重构。
必须在vector里面二分,不能改变原数组的位置。
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
//由块号寻找第一个块元素的下标
#define LL(x) ((x-1)*Len+1)
const int maxn=5e5+5;
LL read()
{
LL x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int a[50050], b[50050], add[505];//b[i]:i的块号
int Len, n;
vector<int> v[505];//每个块的元素
void reset(int x)//重构
{
v[x].clear();
for(int i=LL(x); i<=min(LL(x+1)-1, n); i++)
{
v[x].push_back(a[i]);
}
sort(v[x].begin(), v[x].end());
}
void Add(int L,int R,int x)
{
int bL=b[L], bR=b[R];
if(bL==bR)
{
for(int i=L;i<=R;i++)
{
a[i]+=x;
}
reset(bL);//边块修改完要重构
}
else
{
for(int i=L;i<LL(bL+1);i++)
{
a[i]+=x;
}
reset(bL);//边块修改完要重构
for(int i=bL+1;i<bR;i++)
{
add[i]+=x;
}
for(int i=LL(bR);i<=R;i++)
{
a[i]+=x;
}
reset(bR);//边块修改完要重构
}
}
int query(int L, int R, int x)
{
int ans=0;
int bL=b[L], bR=b[R];
if(bL==bR)
{
for(int i=L;i<=R;i++)
{
if(a[i]+add[bL]<x)
{
ans++;
}
}
}
else
{
for(int i=L;i<LL(bL+1);i++)
{
if(a[i]+add[bL]<x)
{
ans++;
}
}
for(int i=bL+1;i<bR;i++)
{
int f=x-add[i];
ans+=lower_bound(v[i].begin(),v[i].end(), f)-v[i].begin();
}
for(int i=LL(bR);i<=R;i++)
{
if(a[i]+add[bR]<x)
{
ans++;
}
}
}
return ans;
}
void build(int n)
{
Len=n/sqrt(n);
for(int i=1;i<=n;i++)
{
a[i]=read();
add[i]=0;
}
for(int i=1;i<=n;i++)
{
b[i]=(i-1)/Len+1;
v[b[i]].push_back(a[i]);
}
for(int i=1;i<=b[n];i++)
{
sort(v[i].begin(), v[i].end());
}
}
int main()
{
n=read();
build(n);
for(int i=1;i<=n;i++)
{
int op=read(), L=read(), R=read(), c=read();
if(op==0)
{
Add(L, R, c);
}
else
{
printf("%d\n", query(L, R, c*c));
}
}
return 0;
}