LibreOJ #6280. 数列分块入门 4 分块

https://loj.ac/problem/6280
思路：分块$+lazy$标记。和线段树的思路差不多。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
using namespace std;

typedef long long LL;

LL a[50005];//原始序列
int l[2000];//第i个块的左区间端点
int r[2000];//第i个块的右区间端点
LL sum[2000];//第i个块的元素之和
int pos[50005];//第i个点属于第几个块
LL lazy[2000];
int dis;

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=pos[ll]+1;i<=pos[rr]-1;i++)
			lazy[i]+=v;
		for(int i=l[posr];i<=rr;i++)
		{
			a[i]+=v;
			sum[posr]+=v;
		}
	}
}

LL query(int ll,int rr,int c)
{
	int posl=pos[ll];
	int posr=pos[rr];
	if(posl==posr)
	{
		LL temp=0;
		for(int i=ll;i<=rr;i++)
			temp+=a[i]+lazy[posl];
		return temp%(c+1);
	}
	LL temp=0;
	for(int i=ll;i<=r[posl];i++)
		temp+=a[i]+lazy[posl];
	for(int i=pos[ll]+1;i<=pos[rr]-1;i++)
		temp+=sum[i]+lazy[i]*dis;
	for(int i=l[posr];i<=rr;i++)
		temp+=a[i]+lazy[posr];
	return temp%(c+1);
}

int main()
{
	//freopen("a4.in","r",stdin);
	//freopen("a5.out","w",stdout);
	int n;
	scanf("%d",&n);
	dis=sqrt(n);
	int num=ceil(n*1.0/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];
	}
	int t1,t2,t3,t4;
	for(int i=0;i<n;i++)
	{
		scanf("%d %d %d %d",&t1,&t2,&t3,&t4);
		if(t1==0)
			add(t2,t3,t4);
		else
			printf("%lld\n",query(t2,t3,t4));
	}
	return 0;
}