P2801 教主的魔法【分块】

洛谷P2801 -> Click Here

题意

思路

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN = 1e6 + 10;
const int MN = 1e3 +5;
int n,q,tot,sq,L[MN],R[MN],a[MAXN],b[MAXN],add[MN],belong[MAXN];
void Build(){
	sq=sqrt(n);
	tot=n/sq;if(n%sq) tot++;
	for(int i=1;i<=n;i++) belong[i]=(i-1)/sq+1,b[i]=a[i];
	for(int i=1;i<=tot;i++){
		L[i]=(i-1)*sq+1;
		R[i]=sq*i;
	}
	R[tot]=n;
	for(int i=1;i<=tot;i++){
		sort(b+L[i],b+R[i]+1);
	}
	return;
}
void reso(int x){
	for(int i=L[x];i<=R[x];i++){
		b[i]=a[i];
	}
	sort(b+L[x],b+R[x]+1);
}
void Update(int l,int r,int x){
	if(belong[l]==belong[r]){
		for(int i=l;i<=r;i++){
			a[i]+=x;
		}
		reso(belong[l]);
		return;
	}
	for(int i=l;i<=R[belong[l]];i++) a[i]+=x;
	for(int i=L[belong[r]];i<=r;i++) a[i]+=x;
	reso(belong[l]);reso(belong[r]);
	for(int i=belong[l]+1;i<belong[r];i++){
		add[i]+=x;
	}
}
int find(int k,int x){
	int l=L[k],r=R[k],mid;
	while(l<=r){
		mid=l+r>>1;
		b[mid]<x?l=mid+1:r=mid-1;
	}
	return R[k]-l+1;
}
int Query(int l,int r,int x){
	int cnt=0;
	if(belong[l]==belong[r]){
		for(int i=l;i<=r;i++)
			if(a[i]+add[belong[i]]>=x) cnt++;
		return cnt;
	}
	for(int i=l;i<=R[belong[l]];i++)
		if(a[i]+add[belong[i]]>=x) cnt++;
	for(int i=L[belong[r]];i<=r;i++)
		if(a[i]+add[belong[i]]>=x) cnt++;
	for(int i=belong[l]+1;i<belong[r];i++){
		cnt+=find(i,x-add[i]);
	}
	return cnt;
}
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	Build();
	for(int i=1;i<=q;i++){
		char c[6];int x,y,z;
		scanf("%s%d%d%d",&c,&x,&y,&z);
		if(c[0]=='M'){
			Update(x,y,z);
		}else{
			printf("%d\n",Query(x,y,z));
		}
	}
	return 0;
}