线段树总结

杂谈

假设有编号从 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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

幸愉聊信奥

谢谢亲的支持,我会继续努力啦~

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值