线段树区间乘法+区间加法

/*
    线段树区间乘法+区间加法
    by sbn
*/
#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
ll n,m,p,z;
ll arr[100001];
struct node{
        ll l,r;
        ll sum,add,mul;
} tree[400001];
void pushup(int o){
        tree[o].sum=tree[o*2].sum+tree[o*2+1].sum;
        tree[o].sum%=p;
}
inline void add_node(int root,ll num){
    tree[root].add=(tree[root].add+num)%p;
    tree[root].sum=(tree[root].sum+num*(tree[root].r-tree[root].l+1)%p)%p;
}
inline void mul_node(int root,ll num){
    tree[root].mul=(tree[root].mul*num)%p;
    tree[root].sum=(tree[root].sum*num)%p;
    tree[root].add=(tree[root].add*num)%p;
}

void pushdown(ll root){
    ll mid; mid=(tree[root].l+tree[root].r)/2;
    if (tree[root].mul!=1)
        {       mul_node(root*2,tree[root].mul);
                mul_node(root*2+1,tree[root].mul);
                tree[root].mul=1;
        }   
    if (tree[root].add){
                add_node(root*2,tree[root].add);
                add_node(root*2+1,tree[root].add);
                tree[root].add=0;
        }

        return;
}
void jia(int dl,int dr,int root,ll num){
    if (tree[root].l>=dl&&tree[root].r<=dr){
            tree[root].add+=num;
            tree[root].add%=p;
            tree[root].sum+=num*(dr-dl+1);
            tree[root].sum%=p;

            return;
        }
        pushdown(root);
        ll mid=(tree[root].l+tree[root].r)/2;
        if (dr<=mid)    jia(dl,dr,root*2,num);
        else if (dl>mid)    jia(dl,dr,root*2+1,num);
        else {jia(dl,mid,root*2,num);jia(mid+1,dr,root*2+1,num);}
        pushup(root);
}       
void cheng(int dl,int dr,int root,ll num){
        if (tree[root].l==dl&&tree[root].r==dr){
                tree[root].add*=num;
                tree[root].add%=p;
                tree[root].sum*=num;
                tree[root].sum%=p;
                tree[root].mul*=num;
                tree[root].mul%=p;
                return;
        }
        pushdown(root);
        ll mid=(tree[root].l+tree[root].r)/2;
        if (dr<=mid)    cheng(dl,dr,root*2,num);else

        if (dl>mid) cheng(dl,dr,root*2+1,num);
        else    {
        cheng(dl,mid,root*2,num);cheng(mid+1,dr,root*2+1,num);}
        pushup(root);
        return;
}
inline ll getsum(int dl,int dr,int root){
        if (tree[root].l>=dl&&tree[root].r<=dr){
                return tree[root].sum%p;
        }
        pushdown(root);
        ll mid=(tree[root].l+tree[root].r)/2;
        if (dr<=mid)    return getsum(dl,dr,root*2)%p;
        else if (dl>mid) return getsum(dl,dr,root*2+1)%p;
        else return (getsum(dl,dr,root*2)+getsum(dl,dr,root*2+1))%p;
}
void build(int root,int x,int y){
    //cout<<root<<" "<<x<<" "<<y<<endl;
        tree[root].l=x;tree[root].r=y;tree[root].mul=1;tree[root].add=0;
        if (x==y){
            tree[root].sum=arr[x];
            return;
        }
        build(root<<1,x,(x+y)/2);
        build(root<<1|1,(x+y)/2+1,y);
        pushup(root);
}
int main(){
        //freopen("xianduanshu.in","r",stdin);

        cin>>n>>m>>p;
        for(int i=1;i<=n;i++)
            cin>>arr[i];
        build(1,1,n);
        //for (int i=1;i<=2*n;i++)  cout<<tree[i].l<<" "<<tree[i].r<<" "<<tree[i].sum<<endl;    
        for (int i=1;i<=m;i++){
            int a;
            cin>>a;
            if (a==1){
                ll st,ed,v;
                cin>>st>>ed>>v;
                cheng(st,ed,1,v);
            }
            if (a==2){
                ll st,ed,v;
                cin>>st>>ed>>v;
                jia(st,ed,1,v);
                //for (int i=1;i<=2*n;i++)  cout<<tree[i].l<<" "<<tree[i].r<<" "<<tree[i].sum<<endl;
            }
            if (a==3){
                ll st,ed;
                cin>>st>>ed;
                cout<<getsum(st,ed,1)%p<<endl;
            }
        }
        return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值