线段树2

15 篇文章 0 订阅
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<vector>
#include<string.h>
#include<map>
#include<cmath>
#include<queue>
#include<stack>
#define ll long long
#define INF 0x7fffffff
#define MAX 0x3f3f3f3f
#define maxn 100005
#define ull unsigned long long
using namespace std;

ll a[100005],mod;
//树用结构体表示
struct tree{
    int l,r;
    ll add,mul,sum;//add 加法标记,mul乘法标记
}t[400020];
//建树
void bulid(int p,int l,int r){
    t[p].l=l;//p节点维护的区间为l到r
    t[p].r=r;
    t[p].mul=1;//标记初始化
    t[p].add=0;
    if(l==r){//特殊情况l=r 这个区间就只有一个数
        t[p].sum=a[l]%mod;
        return;
    }
    int mid=l+r>>1;
    bulid(p<<1,l,mid);//左子树
    bulid(p<<1|1,mid+1,r);//右子树
    t[p].sum=(t[p<<1].sum+t[p<<1|1].sum)%mod;
}
//pushdown 维护
void pushdown(int p){//和 乘p节点乘法标记,加法标记乘以区间
	t[p<<1].sum=(t[p<<1].sum*t[p].mul+t[p].add*(t[p<<1].r-t[p<<1].l+1))%mod;
	t[p<<1|1].sum=(t[p<<1|1].sum*t[p].mul+t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1))%mod;
	t[p<<1].mul=(t[p<<1].mul*t[p].mul)%mod;//乘法 直接乘p节点乘法标记
	t[p<<1|1].mul=(t[p<<1|1].mul*t[p].mul)%mod;
	t[p<<1].add=(t[p<<1].add*t[p].mul+t[p].add)%mod;//加法 先乘p节点乘法标记再加加法标记
	t[p<<1|1].add=(t[p<<1|1].add*t[p].mul+t[p].add)%mod;
	t[p].add=0;//尾处理
	t[p].mul=1;
	return;
}
//区间乘法
void changeMul(int p,int x,int y,int k){
	if(x<=t[p].l&&t[p].r<=y){
		t[p].add=(t[p].add*k)%mod;
		t[p].mul=(t[p].mul*k)%mod;
		t[p].sum=(t[p].sum*k)%mod;
		return;
	}
	pushdown(p);
	int mid=t[p].l+t[p].r>>1;
	if(x<=mid) changeMul(p<<1,x,y,k);
	if(y>mid) changeMul(p<<1|1,x,y,k);
    t[p].sum=(t[p<<1].sum+t[p<<1|1].sum)%mod;
    return;
}
//区间加法
void changeAdd(int p,int x,int y,int z){
    if(x<=t[p].l&&y>=t[p].r){//要修改的区间覆盖了当前节点,就对其进行修改
        t[p].add=(t[p].add+z)%mod;
        t[p].sum=(t[p].sum+z*(t[p].r-t[p].l+1))%mod;//注意加一
        return;
    }
    pushdown(p);
    int mid=t[p].l+t[p].r>>1;
    if(x<=mid) changeAdd(p<<1,x,y,z);
    if(y>mid) changeAdd(p<<1|1,x,y,z);
    t[p].sum=(t[p<<1].sum+t[p<<1|1].sum)%mod;
    return;
}
//区间查询
ll findd(int p,int x,int y){
    if(x<=t[p].l&&y>=t[p].r)
        return t[p].sum;//如果被覆盖,就返回和
    pushdown(p);
    int mid=t[p].l+t[p].r>>1;
    ll ans=0;
    if(x<=mid) ans=(ans+findd(p<<1,x,y))%mod;//返回左右儿子的和
    if(y>mid) ans=(ans+findd(p<<1|1,x,y))%mod;
    return ans;
}

int main(){
    int n,m,i,q,x,y,z;
    ios::sync_with_stdio(false);
    cin>>n>>m>>mod;
    for(i=1;i<=n;i++)
        cin>>a[i];
    bulid(1,1,n);//头节点编号是1,维护区间是1到n
    for(i=1;i<=m;i++)
    {
        cin>>q>>x>>y;
        if(q==1){
            cin>>z;
            changeMul(1,x,y,z);
        }
        else if(q==2){
            cin>>z;
            changeAdd(1,x,y,z);
        }
        else
            cout<<findd(1,x,y)<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

When_3*four

你的鼓励将是我创作的最大动力

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

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

打赏作者

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

抵扣说明:

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

余额充值