[Ynoi2011]D1T1

题目大意:

给定一个序列$a_1,a_2,\dots,a_n$,进行$m$次操作,每次操作如下:

1. 给定$x,y,z$,对所有下标为$y,y+x,y+2x,\dots$的元素加上$z$(保证$y\leqslant x$)。
2. 给定$l,r$,求$(\sum\limits_{i=l}^r a_i)\bmod{1000000007}$的值。

解题思路:

分块+根号分治简单维护即可。

考虑对$x$的大小分类讨论。

若$x\geqslant \sqrt n$,则我们暴力给每个位置加,需要加的次数为$O(\sqrt n)$次。由于需要查询区间和,用分块维护,总修改、查询复杂度为$O(m\sqrt n)$。

若$x< \sqrt n$,我们需要用另外的方法维护。

注意到单次修改是针对整个序列的元素,所以对$x,y$相同的修改,我们可以累加它的贡献。

那么我们每次查询的时候,需要对于所有的$x$进行一遍查询。故我们需要做到单次$O(1)$。

我们对每个$x$,维护$y$的前缀、后缀和。对于一次询问,我们可以当成把序列分成了若干个大小为$x$的块。中间的整块元素,每个块里肯定所有的$y$都有,增加的贡献就是关于$x$的修改总和。所有块的贡献相同,可以$O(1)$算。边角的话,由于我们记录了前缀、后缀和,也可以$O(1)$算。两个端点在同一个块中,则直接前缀和相减即可。

总时间复杂度$O(m\sqrt n)$。

C++ Code:

#include<iostream>
const int siz=512,M=siz+2,N=2e5+5,Z=N/siz+2,md=1e9+7;
#define bel(x)((x-1)/siz+1)
inline void upd(int&a){a+=a>>31&md;}
int n,blocks,L[Z],R[Z],a[N],sum[Z],m;
int pre[M][M],suf[M][M];
inline int sigma(int l,int r){
	const int bL=bel(l),bR=bel(r);
	int ret=0;
	if(bL==bR)
	for(int i=l;i<=r;++i)upd(ret+=a[i]-md);
	else{
		for(int i=R[bL];i>=l;--i)upd(ret+=a[i]-md);
		for(int i=L[bR];i<=r;++i)upd(ret+=a[i]-md);
		for(int i=bL+1;i<bR;++i)upd(ret+=sum[i]-md);
	}
	return ret;
}
int main(){
	std::ios::sync_with_stdio(0),std::cin.tie(0),std::cout.tie(0);
	std::cin>>n>>m;
	blocks=bel(n);
	for(int i=1;i<=blocks;++i)L[i]=R[i-1]+1,R[i]=i*siz;R[blocks]=n;
	for(int i=1;i<=n;++i)std::cin>>a[i];
	for(int i=1;i<=blocks;++i)
	for(int j=L[i];j<=R[i];++j)upd(sum[i]+=a[j]-md);
	while(m--){
		int op;
		std::cin>>op;
		if(op==1){
			int x,y,z;
			std::cin>>x>>y>>z,z-=md;
			if(x>=siz)
			for(int i=y;i<=n;i+=x)upd(a[i]+=z),upd(sum[bel(i)]+=z);
			else{
				int*pr=pre[x],*sf=suf[x];
				for(int i=x;i>=y;--i)upd(pr[i]+=z);
				for(int i=1;i<=y;++i)upd(sf[i]+=z);
			}
		}else{
			int l,r;
			std::cin>>l>>r;
			int ans=sigma(l,r);
			for(int i=1;i<siz;++i){
				const int bL=(l-1)/i+1,bR=(r-1)/i+1;
				if(bL==bR)upd(ans+=pre[i][(r-1)%i+1]-md),upd(ans-=pre[i][(l-1)%i]);else
				ans=(ans+(bR-bL-1LL)*pre[i][i])%md,upd(ans+=pre[i][(r-1)%i+1]-md),upd(ans+=suf[i][(l-1)%i+1]);
			}
			std::cout<<ans<<'\n';
		}
	}
	return 0;
}

 

posted @ 2019-04-18 18:15  Mrsrz  阅读(274)  评论(0编辑  收藏