前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >浅谈线段树中加与乘标记的下放

浅谈线段树中加与乘标记的下放

作者头像
attack
发布2018-04-10 15:39:02
9360
发布2018-04-10 15:39:02
举报
文章被收录于专栏:数据结构与算法

假设我们一个节点为[val,mul,add],其中val代表该节点的权值,mul为乘法标记,add为加法标记

那么我们有两种表示方式,

  • 第一种:先加再乘

此时该节点为(val+add)*mul

当再遇到一个[_mul,_add]的标记时,

此时节点为[(val+add)*mul+_add]*_mul

把式子展开并重新化为(val+add')*mul'的形式 (也就是提出mul*_mul这一项)得

(val+add+\frac{\_add}{mul})*mul*\_mul

我们发现这里有个除法,会损失很多精度

因此我们换一个思路

  • 第二种:先乘再加

此时该节点为(val*mul)+add

当再遇到一个[\_mul,\_add]的标记时,

此时节点为[(val*mul)+add]*\_mul+\_add

把式子展开并重新化为(val*mul')+add'的形式

val*mul*\_mul+add*\_mul+\_add

我们发现这样不需要除法,因此我们选用第二种

其实线段树标记的下放一般都是这个套路

放一下丑陋的代码

代码语言:javascript
复制
// luogu-judger-enable-o2
// luogu-judger-enable-o2
#include<cstdio>
#include<cmath>
#include<algorithm>
#define ls k<<1
#define rs k<<1|1
#define int long long 
using namespace std;
const int MAXN=1e6+10;
inline int read()
{
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int N,M,mod;
struct node
{
    int mul,add,sum,l,r,siz;
}T[MAXN];
void update(int k)
{
    T[k].sum=(T[ls].sum%mod+T[rs].sum%mod)%mod;
}
void ps(int x,int f)
{
    T[x].mul=(T[x].mul%mod*T[f].mul%mod)%mod;
    T[x].add=(T[x].add*T[f].mul)%mod;
    T[x].add=(T[x].add+T[f].add)%mod;
    T[x].sum=(T[x].sum%mod*T[f].mul%mod)%mod;
    T[x].sum=(T[x].sum+T[f].add%mod*T[x].siz)%mod;
}
void pushdown(int k)
{
    if(T[k].add==0&&T[k].mul==1) return ;
    ps(ls,k);
    ps(rs,k);
    T[k].add=0;
    T[k].mul=1;
}
void Build(int k,int ll,int rr)
{
    T[k].l=ll;T[k].r=rr;T[k].siz=rr-ll+1;T[k].mul=1;
    if(ll==rr)
    {
        T[k].sum=read()%mod;
        return ;
    }
    int mid=ll+rr>>1;
    Build(ls,ll,mid);
    Build(rs,mid+1,rr);
    update(k);
}
void IntervalMul(int k,int ll,int rr,int val)
{
    if(ll<=T[k].l&&T[k].r<=rr)
    {
        T[k].sum=(T[k].sum*val)%mod;
        T[k].mul=(T[k].mul*val)%mod;
        T[k].add=(T[k].add*val)%mod;
        return ;
    }
    pushdown(k);
    int mid=T[k].l+T[k].r>>1;
    if(ll<=mid) IntervalMul(ls,ll,rr,val);
    if(rr>mid)  IntervalMul(rs,ll,rr,val);
    update(k);
}
void IntervalAdd(int k,int ll,int rr,int val)
{
    if(ll<=T[k].l&&T[k].r<=rr)
    {
        T[k].sum=(T[k].sum+T[k].siz*val)%mod;
        T[k].add=(T[k].add+val)%mod;
        return ;
    }
    pushdown(k);
    int mid=T[k].l+T[k].r>>1;
    if(ll<=mid) IntervalAdd(ls,ll,rr,val);
    if(rr>mid)  IntervalAdd(rs,ll,rr,val);
    update(k);
}
int IntervalSum(int k,int ll,int rr)
{
    int ans=0;
    if(ll<=T[k].l&&T[k].r<=rr)
    {
        ans=(ans+T[k].sum)%mod;
        return ans;
    }
    pushdown(k);
    int mid=T[k].l+T[k].r>>1;
    if(ll<=mid) ans=(ans+IntervalSum(ls,ll,rr))%mod;
    if(rr>mid)  ans=(ans+IntervalSum(rs,ll,rr))%mod;
    return ans%mod;
}
main()
{
    #ifdef WIN32
    freopen("a.in","r",stdin);
    #endif
    N=read();M=read();mod=read();
    Build(1,1,N);
    while(M--)
    {
        int opt=read();
        if(opt==1)
        {
            int l=read(),r=read(),val=read()%mod;
            IntervalMul(1,l,r,val);
        }
        else if(opt==2)
        {
            int l=read(),r=read(),val=read()%mod;
            IntervalAdd(1,l,r,val);
        }
        else if(opt==3)
        {
            int l=read(),r=read();
            printf("%lld\n",IntervalSum(1,l,r)%mod);
        }
    }
    return 0;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018-03-17 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档