P3373 【模板】线段树 2
题目描述
如题，已知一个数列，你需要进行下面三种操作：
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的结果。
输入输出样例
说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=8，M<=10
对于70%的数据：N<=1000，M<=10000
对于100%的数据：N<=100000，M<=100000
（数据已经过加强^_^）
样例说明：
故输出应为17、2（40 mod 38=2）
让人心态爆炸的一个模板，调了好久没出来，一怒之下重打了一遍就过了..
主要是注意加与乘的顺序
#include<bits/stdc++.h> #define ll long long #define lson (rt<<1) #define rson (rt<<1|1) using namespace std; const int maxn = 1e5+10; inline int read () { int x=0; char ch=getchar(); while(ch>'9'||ch<'0') ch=getchar(); while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();} return x; } struct node { ll val,add,mul; }tree[maxn<<2]; ll a[maxn],p; int n,m; void pushup(int rt) { tree[rt].val=(tree[lson].val+tree[rson].val)%p; } void pushdown(int l,int r,int rt) { int mid = (l+r)>>1; tree[lson].val=(tree[lson].val*tree[rt].mul+tree[rt].add*(mid-l+1))%p; tree[rson].val=(tree[rson].val*tree[rt].mul+tree[rt].add*(r-mid))%p; tree[lson].mul=(tree[lson].mul*tree[rt].mul)%p; tree[rson].mul=(tree[rson].mul*tree[rt].mul)%p; tree[lson].add=(tree[lson].add*tree[rt].mul+tree[rt].add)%p; tree[rson].add=(tree[rson].add*tree[rt].mul+tree[rt].add)%p; tree[rt].mul=1; tree[rt].add=0; } void buildtree(int l,int r,int rt) { tree[rt].mul=1; if(l==r) { tree[rt].val=a[l]%p; } else { int mid = (l+r)>>1; if(mid>=l) buildtree(l,mid,lson); if(mid+1<=r) buildtree(mid+1,r,rson); pushup(rt); } } void updata1(int x,int y,ll delta,int l,int r,int rt) { if(x<=l&&y>=r) { tree[rt].val=(tree[rt].val*delta)%p; tree[rt].mul=(tree[rt].mul*delta)%p; tree[rt].add=(tree[rt].add*delta)%p; return ; } pushdown(l,r,rt); int mid = (l+r)>>1; if(x<=mid) updata1(x,y,delta,l,mid,lson); if(y>mid) updata1(x,y,delta,mid+1,r,rson); pushup(rt); } void updata2(int x,int y,ll delta,int l,int r,int rt) { if(x<=l&&y>=r) { tree[rt].val=(tree[rt].val+delta*(r-l+1))%p; tree[rt].add+=delta; tree[rt].add%=p; return ; } pushdown(l,r,rt); int mid = (l+r)>>1; if(x<=mid) updata2(x,y,delta,l,mid,lson); if(y>mid) updata2(x,y,delta,mid+1,r,rson); pushup(rt); } ll query(int x,int y,int l,int r,int rt) { if(x<=l&&y>=r) { return tree[rt].val; } pushdown(l,r,rt); int mid = (l+r)>>1; ll ans = 0; if(x<=mid) ans+=query(x,y,l,mid,lson); if(y>mid) ans+=query(x,y,mid+1,r,rson); return ans%p; } int main() { n=read();m=read();scanf("%lld",&p); for(int i=1;i<=n;i++) scanf("%lld",&a[i]); buildtree(1,n,1); while(m--) { int op;scanf("%d",&op); if(op==1) { int x,y;ll z; scanf("%d%d%lld",&x,&y,&z); updata1(x,y,z,1,n,1); } else if(op==2) { int x,y;ll z; scanf("%d%d%lld",&x,&y,&z); updata2(x,y,z,1,n,1); } else { int x,y;scanf("%d%d",&x,&y); printf("%lld\n",query(x,y,1,n,1)%p); } } return 0; }
