bzoj3342 教主的魔法(分块+二分查找)

题意：维护一段序列，两种操作：A.询问[l,r]大于等于v的有多少个。M.[l,r]都加上v

分块练习。。对于区间加法，区间内的整块直接O(1)标记，对于剩下的两头的部分暴力更新。

对于查询：每一块我们都保证是有序的（暴力更新后的块要重构），那么对于区间内的整块，我们直接二分查找，剩下的两头的部分我们暴力统计即可。

#include <cmath>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
#define N 1000005
int n,m,a[N],b[N],block,belong[N],tot=0,add[N];
//a--true value  b--sorted value
inline int read(){
	int 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;
}
inline int min(int x,int y){return x<y?x:y;}
inline void reset(int x){
	int l=(x-1)*block+1,r=min(x*block,n);
	for(int i=l;i<=r;++i) b[i]=a[i];
	sort(b+l,b+r+1);
}
inline void update(int x,int y,int v){
	if(belong[x]==belong[y]){
		for(int i=x;i<=y;++i) a[i]+=v;reset(belong[x]);
	}
	else{
		for(int i=x;i<=block*belong[x];++i) a[i]+=v;
		for(int i=block*(belong[y]-1)+1;i<=y;++i) a[i]+=v;
		for(int i=belong[x]+1;i<belong[y];++i) add[i]+=v;
		reset(belong[x]);reset(belong[y]);//（头尾两块需暴力重构） 
	}
}
inline int query(int x,int y,int v){
	int res=0;
	if(belong[x]==belong[y]){
		for(int i=x;i<=y;++i) if(a[i]+add[belong[i]]>=v) res++;
		return res;
	}
	else{
		for(int i=x;i<=block*belong[x];++i) if(a[i]+add[belong[i]]>=v) res++;
		for(int i=block*(belong[y]-1)+1;i<=y;++i) if(a[i]+add[belong[i]]>=v) res++;
		for(int i=belong[x]+1;i<belong[y];++i){
			int l=(i-1)*block+1,r=i*block;
			int t=lower_bound(b+l,b+r+1,v-add[i])-b;
			res+=r-t+1;
		}
		return res;
	}
}
int main(){
//	freopen("a.in","r",stdin);
	n=read();m=read();block=sqrt(n);
	for(int i=1;i<=n;++i) a[i]=read(),belong[i]=(i-1)/block+1;
	if(n%block) tot=n/block+1;
	else tot=n/block;
	for(int i=1;i<=tot;++i) reset(i);
	while(m--){
		char op[1];int l,r,v;
		scanf("%s",op);l=read();r=read();v=read();
		if(op[0]=='M') update(l,r,v);
		else printf("%d\n",query(l,r,v));
	}
	return 0;
}