1. 程式人生 > >線段樹 區間乘

線段樹 區間乘

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; }

線段樹 區間乘