数列分块入门 4

题目描述

给出一个长为 $n$ 的数列，以及 $n$ 个操作，操作涉及区间加法，区间求和。

输入格式

第一行输入一个数字 $n$。

第二行输入 $n$ 个数字，第 i 个数字为 aia_ia​i​​，以空格隔开。

接下来输入 $n$ 行询问，每行输入四个数字 $\mathrm{o}\mathrm{p}\mathrm{t}$、$l$、$r$、$c$，以空格隔开。

若 $\mathrm{o}\mathrm{p}\mathrm{t}=0$，表示将位于 $[l,r]$ 的之间的数字都加 $c$。

若 $\mathrm{o}\mathrm{p}\mathrm{t}=1$，表示询问位于 $[l,r]$ 的所有数字的和 $mod(c+1)$。

输出格式

对于每次询问，输出一行一个数字表示答案。

#include<iostream>
#include<cstdio>
#include<cmath>
// #include<windows.h>
//#define register int i rii
//#define register int j rij
using namespace std;
long long x[50005],sy[50005],bj[10005],sum[10005],n,c,opt,l,r,cd;
void pushdown(int wz)
{
	for(register int ltt=(wz-1)*cd+1;ltt<=wz*cd;ltt++)
	{
		x[ltt]+=bj[wz];
	}
	bj[wz]=0;
}
void add(int zuo,int you,int v)
{
	int k=sy[zuo];
	int j=sy[you];
	if(k==j)
	{
		pushdown(k);
		for(register int ltt=zuo;ltt<=you;ltt++)
		{
			x[ltt]+=v;
			sum[k]+=v;
		}
		return;
	}
	for(register int ltt=k+1;ltt<=j-1;ltt++)
	{
		bj[ltt]+=v;
		sum[ltt]+=v*cd;
	}
	pushdown(k);
	pushdown(j);
	for(register int ltt=zuo;ltt<=k*cd;ltt++)
	{
		x[ltt]+=v;
		sum[sy[ltt]]+=v;
	}
	for(register int ltt=(j-1)*cd+1;ltt<=you;ltt++)
	{
		x[ltt]+=v;
		sum[sy[ltt]]+=v;
	}
}
long long query(int zuo,int you,int mod)
{
	long long ans=0;
	int k=sy[zuo];
	int j=sy[you];
	if(k==j)
	{
//		pushdown(k);
		for(register int ltt=zuo;ltt<=you;ltt++)
		{
			ans+=x[ltt]+bj[k];
		}
		ans=ans%(mod+1);
		return ans;
	}
	for(register int kkk=k+1;kkk<=j-1;kkk++)
	{
		ans+=sum[kkk];
	}
	for(register int ltt=zuo;ltt<=k*cd;ltt++)
	{
		ans+=x[ltt]+bj[k];
	}
	for(register int ltt=(j-1)*cd+1;ltt<=you;ltt++)
	{
		ans+=x[ltt]+bj[j];
	}
	ans=ans%(mod+1);
	return ans;
}
int main()
{
//	freopen("a1.in","r",stdin);
//	freopen("1.out","w",stdout);
	scanf("%d",&n);
	cd=sqrt(n);
	int cnt=1;
	int bnt=0;
	for(register int i=1;i<=n;i++)
	{
		scanf("%d",&x[i]);
		bnt++;
		sy[i]=cnt;
		sum[cnt]+=x[i];
		if(bnt==cd)
		{
			bnt=0;
			cnt++;
		}
	}
	for(register int i=1;i<=n;i++)
	{
		scanf("%d%d%d%d",&opt,&l,&r,&c);
		if(opt==0)
		{
			add(l,r,c);
		}
		else
		{
			printf("%d\n",query(l,r,c));
		}
	}
//	system("fc 1.out a1.out");
//	system("pause");
}