分块题
构造:每个块排遍序,块大小为
n−−√
加值:给完整的块打标记,不完整的块暴力重构(重新排序)
查询:在每个完整的块中二分查找,不完整的快中暴力判断
代码如下:
#include<algorithm>
#include<ctype.h>
#include<cstdio>
#include<cmath>
#define N 1000050
using namespace std;
inline int read(){
int x=0,f=1;char c;
do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
return x*f;
}
int n,m,Block_size,Block_num,x,y,k;
int a[N],b[N],block[N],v[1020];
char c[5];
inline void reset(int x){
if(x==block[n]) return;
for(int i=Block_size*(x-1)+1;i<=Block_size*x;i++)
b[i]=a[i];
sort(b+Block_size*(x-1)+1,b+Block_size*x+1);
}
inline void Add(int l,int r,int k){
if(block[r]-block[l]<2){
for(int i=l;i<=r;i++)
a[i]+=k;
reset(block[l]);reset(block[r]);
return;
}
else{
for(int i=block[l]+1;i<block[r];i++)
v[i]+=k;
for(int i=l;i<=block[l]*Block_size;i++)
a[i]+=k;
reset(block[l]);
for(int i=r;i>=(block[r]-1)*Block_size+1;i--)
a[i]+=k;
reset(block[r]);
}
}
inline int calc(int l,int r,int k){
int tmp=0,x=r;
while(l<=r){
int mid=l+r>>1;
if(b[mid]>=k) tmp=mid,r=mid-1;
else l=mid+1;
}
if(!tmp) return 0;
return x-tmp+1;
}
inline void Query(int l,int r,int k){
int cnt=0;
if(block[r]-block[l]<2){
for(int i=l;i<=r;i++)
if(a[i]+v[block[i]]>=k) cnt++;
printf("%d\n",cnt);
return;
}
else{
for(int i=block[l]+1;i<block[r];i++){
cnt+=calc(Block_size*(i-1)+1,Block_size*i,k-v[i]);
}
for(int i=l;i<=block[l]*Block_size;i++)
if(a[i]+v[block[i]]>=k) cnt++;
for(int i=r;i>=(block[r]-1)*Block_size+1;i--)
if(a[i]+v[block[i]]>=k) cnt++;
printf("%d\n",cnt);
}
}
main(){
n=read();m=read();
Block_size=sqrt(n);
for(int i=1;i<=n;i++){
a[i]=read();
block[i]=(i-1)/Block_size+1;
}
Block_num=block[n];
for(int i=1;i<Block_num;i++)
reset(i);
while(m--){
scanf("%s",c+1);
x=read();y=read();k=read();
switch(c[1]){
case 'M': Add(x,y,k); break;
case 'A': Query(x,y,k); break;
}
}
return 0;
}