数列分块入门

Posted 2021-11-09 thusloop

数列分块入门
题意：区间修改，区间求和

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=(int)8e18+100;
const int maxn=2e5+100;
int a[maxn],add[maxn];
int pos[maxn],pre[maxn];
int sum[maxn];
int sz;
vector<int>g[maxn];
void update(int l,int r,int val)
{
	for(int i=l; i<=min(r,pos[l]*sz); i++)
	{
		a[i]+=val;
		sum[pos[l]]+=val;
	}
	if(pos[l]!=pos[r])
	{
		for(int i=(pos[r]-1)*sz+1; i<=r; i++)
		{
			a[i]+=val;
			sum[pos[r]]+=val;
		}
		for(int i=pos[l]+1; i<=pos[r]-1; i++)
		{
			add[i]+=val;
		}
	}
}
int ask(int l,int r,int mod)
{
	int ans=0;
	for(int i=l; i<=min(r,pos[l]*sz); i++)
	{
		ans+=a[i]+add[pos[l]];
		ans%=mod;
	}
	if(pos[l]!=pos[r])
	{
		for(int i=(pos[r]-1)*sz+1; i<=r; i++)
		{
			ans+=a[i]+add[pos[r]];
			ans%=mod;
		}
		for(int i=pos[l]+1; i<=pos[r]-1; i++)
		{
			ans+=sz*add[i]+sum[i];
			ans%=mod;
		}
	}
	return ans%mod;
}
signed main()
{
	int n;
	cin>>n;
	sz=sqrt(n);
	for(int i=1; i<=n; i++)
	{
		cin>>a[i];
		pos[i]=(i-1)/sz+1;
		g[pos[i]].pb(a[i]);
		pre[i]=pre[i-1]+a[i];
	}
	for(int i=1;; i++)
	{
		int r=min(i*sz,n);
		sum[i]=pre[r]-pre[(i-1)*sz];
		if(r==n)break;
	}
	for(int __=1; __<=n; __++)
	{
		int op,l,r,c;
		cin>>op>>l>>r>>c;
		if(op==0)
		{
			update(l,r,c);
		}
		else
		{
			cout<<ask(l,r,c+1)<<"\\n";
		}
	}
}