杂谈
假设有编号从 1 到 n 的 n 个点,每个点都存了一些信息,用 [L,R]表示下标从 L 到 R 的这些点。线段树的用处就是,对编号连续的一些点进行修改或者统计操作,修改和统计的复杂度都是 O(log
2
_2
2n)
线段树的原理就是将[1,n]分解成若干特定的子区间(数量不超过 4n ),然后,将每个区间[L,R]都分解为少量特定的子区间,通过对这些少量子区间的修改或者统计,来实现快速对[L,R]的修改或者统计。
由此看出,用线段树统计的东西,必须符合区间加法,否则,不可能通过分成的子区间来得到[L,R]的统计结果。
符合区间加法的例子:
数字之和——总数字之和=左区间数字之和+右区间数字之和
最大值——总最大值=max(左区间最大值,右区间最大值)
代码
区间和、修改(乘,加)
建树
void build(int l,int r,int rt)
{
t[rt].l=l;
t[rt].r=r;
t[rt].mul=1;
t[rt].add=0;
if(l==r)
{
t[rt].sum=begin[l]%mod; // begin 为初始数组.
return ;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
pushup(rt);
}
pushup操作
void pushup(int rt)
{
t[rt].sum=(t[rt<<1].sum+t[rt<<1|1].sum)%mod;
}
pushdown操作
void pushdonw(int rt)
{
int l=t[rt].l,r=t[rt].r,m=(l+r)>>1,lc=rt<<1,rc=rt<<1|1;
t[lc].sum=(t[lc].sum*t[rt].mul%mod+t[rt].add*(m-l+1))%mod;
t[rc].sum=(t[rc].sum*t[rt].mul%mod+t[rt].add*(r-m))%mod;
t[lc].mul=(t[lc].mul*t[rt].mul)%mod;
t[rc].mul=(t[rc].mul*t[rt].mul)%mod;
t[lc].add=(t[lc].add*t[rt].mul+t[rt].add)%mod;
t[rc].add=(t[rc].add*t[rt].mul+t[rt].add)%mod;
t[rt].add=0;
t[rt].mul=1;
return ;
}
在此说明一下,根据我们规定的优先度:
儿子的值=此刻儿子的值 × 爸爸的乘法lazy+儿子长度 × 爸爸的加法lazy,这样进行计算就好了
区间加法
void op_add(int rt,int L,int R,long long c)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return;
if(l>=L&&r<=R)
{
t[rt].add=(t[rt].add+c)%mod;
t[rt].sum=(t[rt].sum+(r-l+1)*c)%mod;
return ;
}
pushdown(rt);
op_add(rt<<1,L,R,c);
op_add(rt<<1|1,L,R,c);
pushup(rt);
return;
}
区间乘法
void op_mul(int rt,int L,int R,long long c)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return;
if(l>=L&&r<=R)
{
t[rt].mul=(t[rt].mul*c)%mod;
t[rt].sum=(t[rt].sum*c)%mod;
t[rt].add=(t[rt].add*c)%mod;
return ;
}
pushdown(rt);
op_mul(rt<<1,L,R,c);
op_mul(rt<<1|1,L,R,c);
pushup(rt);
return;
}
查询区间和
void query(int rt,int L,int R)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return 0;
if(l>=L&&r<=R)return t[rt].sum;
pushdown(rt);
return (query(rt<<1,L,R)+query(rt<<1|1,L,R))%mod;
}
线段树模板
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int MAX=100020,inf=1<<30;
ll mod;
struct tree
{
ll sum,add,mul;
int l,r;
}t[MAX*4];
ll a[MAX];
int n,m;
void pushup(int rt)
{
t[rt].sum=(t[rt<<1].sum+t[rt<<1|1].sum)%mod;
}
void build(int l,int r,int rt)
{
t[rt].l=l;
t[rt].r=r;
t[rt].mul=1;
t[rt].add=0;
if(l==r)
{
t[rt].sum=a[l]%mod;
return ;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
pushup(rt);
}
void pushdown(int rt)
{
int l=t[rt].l,r=t[rt].r;
int m=(l+r)>>1,lc=rt<<1,rc=rt<<1|1;
t[lc].sum=(t[lc].sum*t[rt].mul%mod+t[rt].add*(m-l+1))%mod;
t[rc].sum=(t[rc].sum*t[rt].mul%mod+t[rt].add*(r-m))%mod;
t[lc].mul=(t[lc].mul*t[rt].mul)%mod;
t[rc].mul=(t[rc].mul*t[rt].mul)%mod;
t[lc].add=(t[lc].add*t[rt].mul+t[rt].add)%mod;
t[rc].add=(t[rc].add*t[rt].mul+t[rt].add)%mod;
t[rt].add=0;
t[rt].mul=1;
return ;
}
void op_add(int rt,int L,int R,long long c)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return;
if(l>=L&&r<=R)
{
t[rt].add=(t[rt].add+c)%mod;
t[rt].sum=(t[rt].sum+(r-l+1)*c)%mod;
return ;
}
pushdown(rt);
op_add(rt<<1,L,R,c);
op_add(rt<<1|1,L,R,c);
pushup(rt);
return;
}
void op_mul(int rt,int L,int R,long long c)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return;
if(l>=L&&r<=R)
{
t[rt].mul=(t[rt].mul*c)%mod;
t[rt].sum=(t[rt].sum*c)%mod;
t[rt].add=(t[rt].add*c)%mod;
return ;
}
pushdown(rt);
op_mul(rt<<1,L,R,c);
op_mul(rt<<1|1,L,R,c);
pushup(rt);
return;
}
long long query(int rt,int L,int R)
{
int l=t[rt].l,r=t[rt].r;
if(l>R||r<L)return 0;
if(l>=L&&r<=R)return t[rt].sum;
pushdown(rt);
return (query(rt<<1,L,R)+query(rt<<1|1,L,R))%mod;
}
int main()
{
scanf("%d%d%lld",&n,&m,&mod);
for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
build(1,n,1);
for(int i=1;i<=m;i++)
{
int op;
int x,y;long long k;
scanf("%d",&op);
switch(op)
{
case 1:scanf("%d%d%lld",&x,&y,&k);op_mul(1,x,y,k);break;
case 2:scanf("%d%d%lld",&x,&y,&k);op_add(1,x,y,k);break;
case 3:scanf("%d%d",&x,&y);printf("%lld\n",query(1,x,y));
}
}
return 0;
}