目录
BZOJ1798 维护序列Seq题解ｃｏｄｅ
BZOJ1798 维护序列Seq
题目传送门
题解
比较简单的支持区间加和区间乘的线段树，注意\(pushdown\)的时候区间加和区间乘的处理即可。
ｃｏｄｅ
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=2e5+5;
#define ls(o) o<<1
#define rs(o) (o<<1|1)
int n,m;
ll p;
int a[maxn];
struct node {
int l,r,sz;
ll val;
ll addtag,multag;
}t[maxn<<2];
/*==================Define Area================*/
namespace SegmentTree {
void update(int o) {
t[o].val=(t[ls(o)].val+t[rs(o)].val)%p;
}
void pushdown(int o) {
if(t[o].multag!=1) {
t[ls(o)].multag*=t[o].multag;t[ls(o)].multag%=p;
t[rs(o)].multag*=t[o].multag;t[rs(o)].multag%=p;
t[ls(o)].addtag*=t[o].multag;t[ls(o)].addtag%=p;
t[rs(o)].addtag*=t[o].multag;t[rs(o)].addtag%=p;
t[ls(o)].val*=t[o].multag;t[ls(o)].val%=p;
t[rs(o)].val*=t[o].multag;t[rs(o)].val%=p;
t[o].multag=1;
}
if(t[o].addtag) {
t[ls(o)].addtag+=t[o].addtag;t[ls(o)].addtag%=p;
t[rs(o)].addtag+=t[o].addtag;t[rs(o)].addtag%=p;
t[ls(o)].val+=t[o].addtag*t[ls(o)].sz;t[ls(o)].val%=p;
t[rs(o)].val+=t[o].addtag*t[rs(o)].sz;t[rs(o)].val%=p;
t[o].addtag=0;
}
return ;
}
void Build(int o,int l,int r) {
t[o].l=l,t[o].r=r;t[o].sz=r-l+1;t[o].multag=1;
if(t[o].l==t[o].r) {
t[o].val=a[l];
return ;
}
int mid=(l+r)>>1;
Build(ls(o),l,mid);
Build(rs(o),mid+1,r);
update(o);
}
ll IntervalSum(int o,int l,int r) {
if(t[o].l>r||t[o].r<l) return 0;
if(t[o].l>=l&&t[o].r<=r) {
pushdown(o);
return t[o].val;
}
pushdown(o);
ll ans=0;
int mid=(t[o].l+t[o].r)>>1;
if(mid>=l) ans+=IntervalSum(ls(o),l,r),ans%=p;
if(mid<=r) ans+=IntervalSum(rs(o),l,r),ans%=p;
return ans;
}
void IntervalAdd(int o,int l,int r,int v) {
if(t[o].l>r||t[o].r<l) return ;
if(t[o].l>=l&&t[o].r<=r) {
t[o].val+=t[o].sz*v;
t[o].addtag+=v;
t[o].addtag%=p;
t[o].val%=p;
return ;
}
pushdown(o);
int mid=(t[o].l+t[o].r)>>1;
if(mid>=l) IntervalAdd(ls(o),l,r,v);
if(mid<=r) IntervalAdd(rs(o),l,r,v);
update(o);
return ;
}
void IntervalMul(int o,int l,int r,int v) {
if(t[o].l>r||t[o].r<l) return ;
if(t[o].l>=l&&t[o].r<=r) {
pushdown(o);
t[o].val*=v;t[o].val%=p;
t[o].addtag*=v;t[o].addtag%=p;
t[o].multag*=v;t[o].multag%=p;
return ;
}
pushdown(o);
int mid=(t[o].l+t[o].r)>>1;
if(mid>=l) IntervalMul(ls(o),l,r,v);
if(mid<=r) IntervalMul(rs(o),l,r,v);
update(o);
return ;
}
}
using namespace SegmentTree;
int main() {
read(n);read(p);
for(int i=1;i<=n;i++) {
read(a[i]);
}
read(m);
Build(1,1,n);
for(int i=1;i<=m;i++) {
int opt;
read(opt);
if(opt==1) {
int x,y,k;
read(x);read(y);read(k);
IntervalMul(1,x,y,k);
}
if(opt==2) {
int x,y,k;
read(x);read(y);read(k);
IntervalAdd(1,x,y,k);
}
if(opt==3) {
int x,y;
read(x);read(y);
ll ans=IntervalSum(1,x,y);
printf("%lld\n",ans);
}
}
return 0;
}
转载于:https://www.cnblogs.com/Apocrypha/p/9432790.html
相关资源：JAVA上百实例源码以及开源项目
