[BZOJ5291][线段树]BJOI2018:链上二次求和

版权声明:虽然博主很菜,但是还是请注明出处(我觉得应该没人偷我的博客) https://blog.csdn.net/qq_43346903/article/details/88218669

BZOJ5291

一看就知道要维护一堆前缀和
推了一会发现还有前缀和的前缀和
s u m sum 为前缀和, s s u m ssum 表示前缀和的前缀和
要求 i = l r j = i n s u m ( j ) s u m ( j i ) \sum_{i=l}^r{\sum_{j=i}^{n}}{sum(j)-sum(j-i)}
= i = l r ( s s u m ( n ) s s u m ( i 1 ) s s u m ( l i ) ) =\sum_{i=l}^r(ssum(n)-ssum(i-1)-ssum(l-i))
拆开发现之和区间ssum的和有关
考虑维护这个东西,区间加一个数实际上对sum是一次函数型的影响,所以对ssum是二次函数型的影响,那么就是线段树维护二次函数的基操了

Code:

#include<bits/stdc++.h>
#define ll long long
#define mid ((l+r)>>1)
#define mod 1000000007
#define inv2 500000004
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=2e5+5;
inline ll dec(ll x){if(x>=mod) x-=mod;return x;}
int n,m;
ll a[N],s1[N],s2[N];
namespace segtree{
	struct seg{ll x1,x2,x3,sum,a,b,c;}tr[N<<2];
	inline void pushup(int k){tr[k].sum=(tr[k<<1].sum+tr[k<<1|1].sum)%mod;}
	inline void pushnow(int k,ll a,ll b,ll c){
		tr[k].a+=a,tr[k].b+=b,tr[k].c+=c;
		(tr[k].sum+=a*tr[k].x1+b*tr[k].x2+c*tr[k].x3)%=mod;
		if(tr[k].a>=mod) tr[k].a-=mod;
		if(tr[k].b>=mod) tr[k].b-=mod;
		if(tr[k].c>=mod) tr[k].c-=mod;
	}
	inline void pushdown(int k){
		if(tr[k].a || tr[k].b || tr[k].c){
			pushnow(k<<1,tr[k].a,tr[k].b,tr[k].c),pushnow(k<<1|1,tr[k].a,tr[k].b,tr[k].c);
			tr[k].a=tr[k].b=tr[k].c=0;
		}
	}
	inline void build(int k,int l,int r){
		tr[k].x3=r-l+1,tr[k].x2=s1[r]-(l?s1[l-1]:0),tr[k].x1=s2[r]-(l?s2[l-1]:0);
		if(l==r){tr[k].sum=a[l];return;}
		build(k<<1,l,mid),build(k<<1|1,mid+1,r),pushup(k);
	}
	inline void modify(int k,int l,int r,int ql,int qr,ll a,ll b,ll c){
		if(ql>r || qr<l)return;
		if(ql<=l && r<=qr)return pushnow(k,a,b,c);
		pushdown(k);
		if(qr<=mid) modify(k<<1,l,mid,ql,qr,a,b,c);
		else if(ql>mid) modify(k<<1|1,mid+1,r,ql,qr,a,b,c);
		else modify(k<<1,l,mid,ql,mid,a,b,c),modify(k<<1|1,mid+1,r,mid+1,qr,a,b,c);
		pushup(k);
	}
	inline ll query(int k,int l,int r,int ql,int qr){
		if(ql>r || qr<l) return 0;
		if(ql<=l && r<=qr) return tr[k].sum;
		pushdown(k);
		if(qr<=mid) return query(k<<1,l,mid,ql,qr);
		if(ql>mid) return query(k<<1|1,mid+1,r,ql,qr);
		return (query(k<<1,l,mid,ql,mid)+query(k<<1|1,mid+1,r,mid+1,qr))%mod;
	}	
}
using namespace segtree;
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;++i) a[i]=(read()+a[i-1])%mod,s1[i]=(s1[i-1]+i)%mod,s2[i]=(s2[i-1]+(ll)i*i%mod)%mod;
	for(int i=1;i<=n;++i) (a[i]+=a[i-1])%=mod;
	build(1,1,n);
	while(m--){
		int op=read(),l=read(),r=read(),len;
		ll v;
		if(l>r) swap(l,r);
		if(op==1){
			v=read(),len=r-l+1;
			ll a=v*inv2%mod,b=v*(3-2*l+mod)%mod*inv2%mod,c=v*(l-1)%mod*(l-2)%mod*inv2%mod;
			modify(1,1,n,l,r,a,b,c);
			if(r==n) continue;
			a=0,b=v*len%mod,c=(((ll)(len+1)*len/2%mod*v-v*len%mod*r)%mod+mod)%mod;
			modify(1,1,n,r+1,n,a,b,c);
		}
		else{
			l=max(l,1);
			ll s1=query(1,1,n,n,n)*(r-l+1)%mod,s2=query(1,1,n,max(l-1,1),r-1),s3=query(1,1,n,max(n-r,1),n-l);
			cout<<(s1-s2-s3+mod*3)%mod<<"\n";
		}
	}
}

猜你喜欢

转载自blog.csdn.net/qq_43346903/article/details/88218669