線段樹 區間乘
阿新 • • 發佈:2017-08-06
http fin sum class ++ col 數列 其中 輸入輸出格式
題目描述
如題,已知一個數列,你需要進行下面兩種操作:
1.將某區間每一個數加上x
2.將某區間每一個數乘上x
3.求出某區間每一個數的和
輸入輸出格式
輸入格式:
第一行包含三個整數N、M、P,分別表示該數列數字的個數、操作的總個數和模數。
第二行包含N個用空格分隔的整數,其中第i個數字表示數列第i項的初始值。
接下來M行每行包含3或4個整數,表示一個操作,具體如下:
操作1: 格式:1 x y k 含義:將區間[x,y]內每個數乘上k
操作2: 格式:2 x y k 含義:將區間[x,y]內每個數加上k
操作3: 格式:3 x y 含義:輸出區間[x,y]內每個數的和對P取模所得的結果
輸出格式:
輸出包含若幹行整數,即為所有操作3的結果。
輸入輸出樣例
輸入樣例#1:5 5 38 1 5 4 2 3 2 1 4 1 3 2 5 1 2 4 2 2 3 5 5 3 1 4輸出樣例#1:
17 2
說明
時空限制:1000ms,128M
數據規模:
對於30%的數據:N<=8,M<=10
對於70%的數據:N<=1000,M<=10000
對於100%的數據:N<=100000,M<=100000
(數據已經過加強^_^)
樣例說明:
一個小錯誤改了一上午
#include<iostream> #include<queue> #include<cstdio> #include<math.h> #include<cstring> #include<algorithm> using namespace std; #define LL long long long long n,Q; long long ans,P; long long sum[400009]; long long dx[400009],ch[400009]; long long a[400009]; void build(LL l,LL r,LL id) { ch[id]=1;sum[id]=0; if(l==r) {sum[id]=a[l]%P;return;} int mid=(l+r)/2; build(l,mid,id*2);build(mid+1,r,id*2+1); sum[id]= (sum[id*2] + sum[id*2+1])%P; return; } void cheng(LL l,LL r,LL id,LL tl,LL tr,LL x) { if(tl<=l&&r<=tr) { ch[id]=((LL)ch[id]*x)%P; dx[id]=((LL)dx[id]*x)%P; sum[id]=((LL)sum[id]*x)%P; return; } int mid=(l+r)/2; if((ch[id]!=1) || dx[id]) { dx[id<<1]=((LL)dx[id<<1]*ch[id]+dx[id])%P; ch[id<<1]=((LL)ch[id<<1]*ch[id])%P; sum[id<<1]=(((LL)sum[id<<1]*ch[id])%P+(LL)dx[id]*(mid-l+1)%P)%P; ch[id<<1|1]=((LL)ch[id<<1|1]*ch[id])%P;dx[id<<1|1]=((LL)dx[id<<1|1]*ch[id]+dx[id])%P; sum[id<<1|1]=(((LL)sum[id<<1|1]*ch[id])%P+(LL)dx[id]*(r-mid)%P)%P; ch[id]=1;dx[id]=0; } if(tl<=mid) cheng(l,mid,id*2,tl,tr,x); if(tr>=mid+1) cheng(mid+1,r,id*2+1,tl,tr,x); sum[id]=(sum[id<<1]+sum[id<<1|1])%P; return ; } void add(LL l,LL r,LL id,LL tl,LL tr,LL x) { if(tl<=l&&r<=tr) { (dx[id]+=x)%P; (sum[id]+=(LL)(r-l+1)*x)%P; return; } int mid=(l+r)/2; if( dx[id] || ( ch[id]!=1 )) { ch[id<<1]=((LL)ch[id<<1]*ch[id])%P;dx[id<<1]=((LL)dx[id<<1]*ch[id]+dx[id])%P; sum[id<<1]=(((LL)sum[id<<1]*ch[id])%P+(LL)dx[id]*(mid-l+1)%P)%P; ch[id<<1|1]=((LL)ch[id<<1|1]*ch[id])%P;dx[id<<1|1]=((LL)dx[id<<1|1]*ch[id]+dx[id])%P; sum[id<<1|1]=(((LL)sum[id<<1|1]*ch[id])%P+(LL)dx[id]*(r-mid)%P)%P; ch[id]=1;dx[id]=0; } if(tl<=mid) add(l,mid,id*2,tl,tr,x); if(tr>=mid+1) add(mid+1,r,id*2+1,tl,tr,x); sum[id]=(sum[id<<1]+sum[id<<1|1])%P; return ; } long long ask(LL l,LL r,LL id,LL tl,LL tr) { if(tl<=l&&r<=tr) return sum[id]%P; int mid=(l+r)/2; if( dx[id] || ( ch[id]!=1 )) { ch[id<<1]=((LL)ch[id<<1]*ch[id])%P;dx[id<<1]=((LL)dx[id<<1]*ch[id]+dx[id])%P; sum[id<<1]=(((LL)sum[id<<1]*ch[id])%P+(LL)dx[id]*(mid-l+1)%P)%P; ch[id<<1|1]=((LL)ch[id<<1|1]*ch[id])%P;dx[id<<1|1]=((LL)dx[id<<1|1]*ch[id]+dx[id])%P; sum[id<<1|1]=(((LL)sum[id<<1|1]*ch[id])%P+(LL)dx[id]*(r-mid)%P)%P; ch[id]=1;dx[id]=0; } long long ans=0; if(tl <= mid) ans=(ans+ask(l,mid,id*2,tl,tr))%P; if(tr >= mid+1) ans=(ans+ask(mid+1,r,id*2+1,tl,tr))%P; return ans%P; } int main() { scanf("%lld%lld%lld",&n,&Q,&P); for(int i=1;i<=n;i++) scanf("%lld",&a[i]); build(1,n,1); for(LL i=1,A,l,r,x;i<=Q;i++) { scanf("%lld",&A); if(A==1) { scanf("%lld%lld%lld",&l,&r,&x); cheng(1,n,1,l,r,x); }else if(A==2) { scanf("%lld%lld%lld",&l,&r,&x); add(1,n,1,l,r,x); }else { scanf("%lld%lld",&l,&r); cout<<ask(1,n,1,l,r)%P<<endl; } } return 0; }
線段樹 區間乘