线段树之:【区间修改】【多懒标记问题】【多懒标记的关系处理】【pushdown()分离出eval()函数的妙用】

一.pushdown()分离出eval()函数的妙用

 

 本题分析(多懒标记的处理)

 

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long LL;
int n,m; //n是序列长度,m是操作数个数
int p;   //p是取模的数
int w[N];
struct Node
{
    int l,r;
    LL sum,add,mul;
}tr[N*4];

void pushup(int u)
{
    tr[u].sum=(tr[u<<1].sum+tr[u<<1|1].sum)%p;
}


//处理pushdown中又加又乘的东西
void eval(Node &t,int add,int mul)
{
    //首先是非懒标记的变化:先乘后加
    t.sum=(LL)(t.sum*mul+(LL)(t.r-t.l+1)*add)%p;
    //其次是懒标记的变化
    t.mul=(LL)t.mul*mul%p;
    t.add=((LL)t.add*mul+add)%p;
}

void pushdown(int u)
{
    auto &root=tr[u];
    eval(tr[u<<1],root.add,root.mul);
    eval(tr[u<<1|1],root.add,root.mul);
    
    root.add=0,root.mul=1;
}




void build(int u,int l,int r)
{
    tr[u]={l,r,0,0,1};
    if(l==r)
    {
        tr[u]={l,r,w[r],0,1};
        return;
    }
    
    int mid=l+r>>1;
    build(u<<1,l,mid),build(u<<1|1,mid+1,r);
    
    pushup(u);
}


void modify(int u,int l,int r,int add,int mul)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        eval(tr[u],add,mul);  //修改根节点的懒标记和非懒标记
    }
    else
    {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid)
        {
            modify(u<<1,l,r,add,mul);
        }
        if(r>mid)
        {
            modify(u<<1|1,l,r,add,mul);
        }
        pushup(u);
    }
    
}

LL query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        return tr[u].sum;
    }
    
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    LL sum=0;
    if(l<=mid) sum=query(u<<1,l,r);
    if(r>mid) sum=(sum+query(u<<1|1,l,r))%p;
    
    return sum;
}


int main()
{
    scanf("%d%d",&n,&p);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&w[i]);
    }
    
    build(1,1,n);
    
    scanf("%d",&m);
    while(m--)
    {
        int t,l,r,d;
        scanf("%d%d%d",&t,&l,&r);
        
        if(t==1)
        {
            scanf("%d",&d);
            modify(1,l,r,0,d);
        }
        else if(t==2)
        {
            scanf("%d",&d);
            modify(1,l,r,d,1);
        }
        else printf("%d\n",query(1,l,r));
    }
    
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值