【洛谷p2801】教主的魔法【分块模板】

给出一个数列,支持区间修改,区间查询大于等于key的值个数,分块模板加一个小二分。

详情见代码。A指查询(L到R大于x),M指修改(将L到R增加x)

#include<bits/stdc++.h>
#define in in()
using namespace std;
int in{
	int cnt=0,f=1; char ch=0;
	while(!isdigit(ch)){
		ch=getchar();
		if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}
	return cnt*f;
}
int n,q;
int belong[1000003],l[1000003],r[1000003],p[1000003];
int d[1000003];
int a[1000003];
void build(){
	int block=sqrt(n);
	int num=n/block;
	if(n%block)num++;
	for(int i=1;i<=n;i++){
		belong[i]=(i-1)/block+1;
		d[i]=a[i];
	}
	for(int i=1;i<=num;i++){
		l[i]=(i-1)*block+1;
		r[i]=i*block;
	}
	r[num]=n;
	for(int i=1;i<=num;i++){
		sort(d+l[i],d+r[i]+1);
	}
}
void update(int L,int R,int v){
	if(belong[L]==belong[R]){
		for(int i=L;i<=R;i++){
			a[i]+=v;
		}
		for(int i=l[belong[L]];i<=r[belong[L]];i++){
			d[i]=a[i];
		}
		sort(d+l[belong[L]],d+r[belong[L]]+1);
	}
	else{
		for(int i=L;i<=r[belong[L]];i++){
			a[i]+=v;
		}
		for(int i=l[belong[L]];i<=r[belong[L]];i++){
			d[i]=a[i];
		}
		sort(d+l[belong[L]],d+r[belong[L]]+1);
		
		for(int i=l[belong[R]];i<=R;i++){
			a[i]+=v;
		}
		for(int i=l[belong[R]];i<=r[belong[R]];i++){
			d[i]=a[i];
		}
		sort(d+l[belong[R]],d+r[belong[R]]+1);
		
		for(int i=belong[L]+1;i<=belong[R]-1;i++){
			p[i]+=v;
		}
	}
}
int query(int L,int R,int v){
	int ans=0;
	if(belong[L]==belong[R]){
		for(int i=L;i<=R;i++){
			if(a[i]+p[belong[i]]>=v)ans++;
		}
		return ans;
	}
	for(int i=L;i<=r[belong[L]];i++){
		if(a[i]+p[belong[i]]>=v)ans++;
	}
	for(int i=l[belong[R]];i<=R;i++){
		if(a[i]+p[belong[i]]>=v)ans++;
	}
	for(int i=belong[L]+1;i<=belong[R]-1;i++){
		int ll=l[i],rr=r[i],mid,result=0;;
		while(ll<=rr){
			mid=(ll+rr)>>1;
			if(d[mid]+p[i]>=v)rr=mid-1,result=r[i]-mid+1;
			else ll=mid+1;
		}
		ans+=result;
	}
	return ans;
}
int main(){
	//freopen("testdata.in","r",stdin);
//	freopen("1.out","w",stdout);
	n=in;q=in;
	for(int i=1;i<=n;i++){
		a[i]=in;
	}
	build();
	int x,y,z;char c[6];
	while(q--){
		scanf("%s",c);x=in;y=in;z=in;
		if(c[0]=='A'){
			printf("%d\n",query(x,y,z));
		}
		else update(x,y,z);
	} 
	return 0;
}

本代码重点在于随时维护一个d数组保持其按照每块块内升序,这样在查询大块大于时可以用二分确定个数。

代码细节很多,一定要记清楚belong对应的意义和l,r对应的意义。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值