1. 程式人生 > 實用技巧 >【UOJ 56】線段樹區間加和乘

【UOJ 56】線段樹區間加和乘

【題目描述】:

如題,已知一個數列,你需要進行下面兩種操作:

1.將某區間每一個數加上x

2.將某區間每一個數乘上x

3.求出某區間每一個數的和

【輸入描述】:

第一行包含三個整數N、M、P,分別表示該數列數字的個數、操作的總個數和模數。

第二行包含N個用空格分隔的整數,其中第i個數字表示數列第i項的初始值。

接下來M行每行包含3或4個整數,表示一個操作,具體如下:

操作1: 格式:1 x y k 含義:將區間[x,y]內每個數乘上k(>0)

操作2: 格式:2 x y k 含義:將區間[x,y]內每個數加上k(>0)

操作3: 格式:3 x y 含義:輸出區間[x,y]內每個數的和對P取模所得的結果

【輸出描述】:

輸出包含若干行整數,即為所有操作3的結果。

【樣例輸入】:

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

【樣例輸出】:

17
2

【時間限制、資料範圍及描述】:

時間:1s 空間128M

對於30%的資料:N<=10,M<=10

對於70%的資料:N<=1,000,M<=10,000

對於100%的資料:N<=100,000,M<=100,000

題解:新學區間乘操作 √

#include<cstdio>
#include<iostream>
#include<cmath>
#include
<cstdlib> #include<cstring> #include<algorithm> #include<bits/stdc++.h> typedef long long ll; using namespace std; const int N=100003; int yc,n,T,x,y; ll z,mod,tim[N*4]; ll a[N*4],sum[N*4],add[N*4]; void pushup(int rt) { sum[rt]=(sum[rt*2]+sum[rt*2+1])%mod; } void build(int l,int
r,int rt){ tim[rt]=1; add[rt]=0; if(l==r){ sum[rt]=a[l]; return ; } int m=(l+r)/2; build(l,m,rt*2); build(m+1,r,rt*2+1); pushup(rt); } void pushdown(int rt,int ln,int rn){ if(add[rt]==0 && tim[rt]==1) return; if(ln+rn==1) return ; sum[rt*2]=(sum[rt*2]*tim[rt]+add[rt]*ln)%mod; sum[rt*2+1]=(sum[rt*2+1]*tim[rt]+add[rt]*rn)%mod; add[rt*2]=(add[rt*2]*tim[rt]+add[rt])%mod; add[rt*2+1]=(add[rt*2+1]*tim[rt]+add[rt])%mod; add[rt]=0; tim[rt*2]=(tim[rt*2]*tim[rt])%mod; tim[rt*2+1]=(tim[rt*2+1]*tim[rt])%mod; tim[rt]=1; } void update1(int L,int R,ll c,int l,int r,int rt){ int m=(l+r)/2; pushdown(rt,m-l+1,r-m); if(L<=l && r<=R) { sum[rt]=(sum[rt]*c)%mod; tim[rt]*=c; return ; } if(L<=m) update1(L,R,c,l,m,rt*2); if(R>m) update1(L,R,c,m+1,r,rt*2+1); pushup(rt); } void update2(int L,int R,ll c,int l,int r,int rt){ int m=(l+r)/2; pushdown(rt,m-l+1,r-m); if(L<=l && r<=R) { sum[rt]=(sum[rt]+c*(r-l+1))%mod; add[rt]+=c; return ; } if(L<=m) update2(L,R,c,l,m,rt*2); if(R>m) update2(L,R,c,m+1,r,rt*2+1); pushup(rt); } ll query(int L,int R,int l,int r,int rt){ if(L<=l && r<=R) return sum[rt]; int m=(l+r)/2; pushdown(rt,m-l+1,r-m); ll ans=0; if(L<=m) ans=(ans+query(L,R,l,m,rt*2))%mod; if(R>m) ans=(ans+query(L,R,m+1,r,rt*2+1))%mod; return ans; } int main(){ scanf("%d %d %d",&n,&T,&mod); for(int i=1;i<=n;i++) tim[i]=1; for(int i=1;i<=n;i++) scanf("%lld",&a[i]); build(1,n,1); while(T--){ scanf("%d",&yc); if(yc==1){ scanf("%d %d %lld",&x,&y,&z); update1(x,y,z,1,n,1); } if(yc==2){ scanf("%d %d %lld",&x,&y,&z); update2(x,y,z,1,n,1); } if(yc==3){ scanf("%d %d",&x,&y); cout<<query(x,y,1,n,1)<<endl; } } return 0; }