线段树2.2——区间修改2

最新推荐文章于 2024-05-21 21:07:34 发布

集304

最新推荐文章于 2024-05-21 21:07:34 发布

阅读量747

点赞数 31

文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_73702550/article/details/136334503

版权

请添加图片描述

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
int T;
int a[N];
int n,mod;
struct info
{
    int mul;
    int add;
};
struct node
{
    int len,val;
    info t;
}tr[N*4];
info operator+(const info &l,const info &r)
{
    return {l.mul * r.mul % mod,(l.add*r.mul+r.add)%mod};
}
void settag(int p,info tag)
{ 
   tr[p].t=tr[p].t+tag;
   tr[p].val=(tr[p].val*tag.mul+tag.add*tr[p].len)%mod;
}
void pushdown(int p)
{
    info tag=tr[p].t;
    if(tag.mul!=1||tag.add!=0)
    {
       settag(2*p,tag);
       settag(2*p+1,tag);
       tr[p].t={1,0};
    }
}
void update(int p)
{
   tr[p].val=(tr[2*p].val+tr[2*p+1].val)%mod;
}
void build(int p,int l,int r)
{
    tr[p].len=r-l+1;
    tr[p].t={1,0};
    if(l==r){
        tr[p].val=a[l];
        return ;
    }else{
        int mid=(l+r)/2;
        build(2*p,l,mid);
        build(2*p+1,mid+1,r);
        update(p);
    }
}
void modify(int p,int l,int r,int ql,int qr,info tag)
{
    if(ql==l&&qr==r){
        settag(p,tag);
        return ;
    }
    pushdown(p);
    int mid=(l+r)/2;
    if(qr<=mid) modify(2*p,l,mid,ql,qr,tag);
    else if(ql>mid) modify(2*p+1,mid+1,r,ql,qr,tag);
    else
    {
        modify(2*p,l,mid,ql,mid,tag);
        modify(2*p+1,mid+1,r,mid+1,qr,tag);
    }
    update(p);
}
int query(int p,int l,int r,int ql,int qr)
{
    if(ql==l&&qr==r){
        return tr[p].val;
    }
    pushdown(p);
    int mid=(l+r)/2;
    if(qr<=mid) return query(2*p,l,mid,ql,qr);
    else if(ql>mid) return query(2*p+1,mid+1,r,ql,qr);
    else return (query(2*p,l,mid,ql,mid)+query(2*p+1,mid+1,r,mid+1,qr))%mod;
    update(p);
}
signed main()
{
    //ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
     scanf("%lld%lld", &n, &mod);
    for (int i = 1; i <= n; i++)
        scanf("%lld", &a[i]);
    int q;
    cin>>q;
    build(1, 1, n);
    for (int i = 0; i < q; i++) {
        int ty;
        cin>>ty;
        if(ty<=2){
         int l, r, d;
         cin>>l>>r>>d;
         if (ty == 2) modify(1, 1, n, l, r, (info){1, d});
         else if (ty == 1) modify(1, 1, n, l, r, (info){d, 0});
        }
        else {
            int l, r;
            cin>>l>>r;
            auto ans = query(1, 1, n, l, r);
            printf("%lld\n", ans);
        }
    }
    return 0;
}