数列分块入门 4 LibreOJ - 6280-------------------分块

数列分块入门 4 LibreOJ - 6280-------------------分块_第1张图片
数列分块入门 4 LibreOJ - 6280-------------------分块_第2张图片
解析:

求区间和需要另开一个sum数组来记录每一个分块的总和

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+1000;
typedef long long ll;
int l[N];
int r[N];
int pos[N];
ll lazy[N];
ll sum[N];
ll a[N],res;
int n;
int op,u,v,c,dis,num;
void add(int ll,int rr,int v)
{
	int posl=pos[ll];//左区间端点属于第几个块
	int posr=pos[rr];//右区间端点属于第几个块
	if(posl==posr)//暴力修改相同块的区间 
	{
		for(int i=ll;i<=rr;i++) a[i]+=v,sum[posl]+=v;
	 } 
	 else
	 {
	 	for(int i=ll;i<=r[posl];i++) a[i]+=v,sum[posl]+=v;
		 for(int i=l[posr];i<=rr;i++) a[i]+=v,sum[posr]+=v;
		 for(int i=pos[ll]+1;i<=pos[rr]-1;i++) lazy[i]+=v ;
	 }
}
ll query(int ll,int rr,int MOD)
{
	int posl=pos[ll];
	int posr=pos[rr];
	res=0;
	if(posl==posr)
	{
		for(int i=ll;i<=rr;i++) res=(res+a[i]+lazy[posl])%MOD;
		return res%MOD;
	}
	for(int i=ll;i<=r[posl];i++) res=(res+a[i]+lazy[posl])%MOD;
	for(int i=l[posr];i<=rr;i++) res=(res+a[i]+lazy[posr])%MOD;
	for(int i=pos[ll]+1;i<=pos[rr]-1;i++ ) res=(res+sum[i]+lazy[i]*dis);
	return res%MOD;
}
int main()
{
	scanf("%d",&n);
	dis=sqrt(n);
	num=ceil(1.0*n/dis);
	for(int i=1;i<=num;i++) l[i]=(i-1)*dis+1 ,r[i]=i*dis;
	r[num]=n;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		pos[i]=(i-1)/dis+1;
		sum[pos[i]]+=a[i];
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d %d %d",&op,&u,&v,&c);
		if(op==0) add(u,v,c);
		else cout<<query(u,v,c+1)<<endl;
	}
}

你可能感兴趣的:(分块)