BZOJ1798:[AHOI2009]維護序列
阿新 • • 發佈:2018-11-12
淺談樹狀陣列與線段樹:https://www.cnblogs.com/AKMer/p/9946944.html
題目傳送門:https://www.lydsy.com/JudgeOnline/problem.php?id=1798
線段樹區間加區間乘區間詢問裸題。因為乘標記會影響加標記,所以優先下傳乘標記。
時間複雜度:\(O(mlogn)\)
空間複雜度:\(O(n)\)
程式碼如下:
#include <cstdio> using namespace std; const int maxn=1e5+5; int n,m,pps; int a[maxn]; int read() { int x=0,f=1;char ch=getchar(); for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1; for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0'; return x*f; } struct segment_tree { int tree[maxn<<2],add_tag[maxn<<2],mul_tag[maxn<<2]; void updata(int p) { tree[p]=(tree[p<<1]+tree[p<<1|1])%pps; } void build(int p,int l,int r) { if(l==r) { tree[p]=a[l]; return; }mul_tag[p]=1,add_tag[p]=0;//乘1加0等於本身 int mid=(l+r)>>1; build(p<<1,l,mid); build(p<<1|1,mid+1,r); updata(p); } void add_add_tag(int p,int l,int r,int v) { tree[p]=(tree[p]+1ll*(r-l+1)*v%pps)%pps; add_tag[p]=(add_tag[p]+v)%pps; } void add_mul_tag(int p,int l,int r,int v) { tree[p]=1ll*tree[p]*v%pps; add_tag[p]=1ll*add_tag[p]*v%pps; mul_tag[p]=1ll*mul_tag[p]*v%pps; } void push_down(int p,int l,int r) { int mid=(l+r)>>1; if(mul_tag[p]!=1) { add_mul_tag(p<<1,l,mid,mul_tag[p]); add_mul_tag(p<<1|1,mid+1,r,mul_tag[p]); mul_tag[p]=1; } if(add_tag[p]) { add_add_tag(p<<1,l,mid,add_tag[p]); add_add_tag(p<<1|1,mid+1,r,add_tag[p]); add_tag[p]=0; } } void change_mul(int p,int l,int r,int L,int R,int v) { if(L<=l&&r<=R) { add_mul_tag(p,l,r,v); return; } int mid=(l+r)>>1;push_down(p,l,r); if(L<=mid)change_mul(p<<1,l,mid,L,R,v); if(R>mid)change_mul(p<<1|1,mid+1,r,L,R,v); updata(p); } void change_add(int p,int l,int r,int L,int R,int v) { if(L<=l&&r<=R) { add_add_tag(p,l,r,v); return; } int mid=(l+r)>>1;push_down(p,l,r); if(L<=mid)change_add(p<<1,l,mid,L,R,v); if(R>mid)change_add(p<<1|1,mid+1,r,L,R,v); updata(p); } int query(int p,int l,int r,int L,int R) { if(L<=l&&r<=R)return tree[p]; int mid=(l+r)>>1,res=0;push_down(p,l,r); if(L<=mid)res=query(p<<1,l,mid,L,R); if(R>mid)res=(res+query(p<<1|1,mid+1,r,L,R))%pps; return res; } }T; int main() { n=read(),pps=read(); for(int i=1;i<=n;i++) a[i]=read()%pps; T.build(1,1,n);m=read(); for(int i=1;i<=m;i++) { int opt=read(),l=read(),r=read(),v; if(opt!=3)v=read(); if(opt==1)T.change_mul(1,1,n,l,r,v); else if(opt==2)T.change_add(1,1,n,l,r,v); else printf("%d\n",T.query(1,1,n,l,r)); } return 0; }