洛谷 P3373 线段树(用线段树维护,考虑加和乘的顺序)

题目链接https://www.luogu.com.cn/problem/P3373
在这里插入图片描述
在这里插入图片描述
加和乘的顺序不同会对结果造成影响。

1.(a+b)*c=a*c+b*c;
2.a*c+b
结果是不同的

处理办法:添加乘法标记时,把加法标记*c即可。
需要mul和add两个标记分别标记乘法和加法
懒惰标记下传时
add更新为:add * 父亲的mul+父亲的add
mul更新为:mul * 父亲的mul
代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=4*1e5+10;
LL p;
LL sum[N],add[N],add1[N];
void build(LL l,LL r,LL o)/*先纵后横*/
{
    add[o]=0;
    add1[o]=1;
    if(l==r)
    {
        scanf("%lld",&sum[o]);
        sum[o]%=p;
        return;
    }
    LL mid=(l+r)>>1;
    build(l,mid,o<<1);
    build(mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%p;
}
void pushdown(LL len,LL o)
{	
	add1[o<<1]=(add1[o<<1]*add1[o])%p;
        add1[o<<1|1]=(add1[o<<1|1]*add1[o])%p;
        
       add[o<<1]=(add[o<<1]*add1[o]+add[o])%p;
       add[o<<1|1]=(add[o<<1|1]*add1[o]+add[o])%p;
    
      sum[o<<1]=(add1[o]*sum[o<<1]+add[o]*(len-(len>>1)))%p;
      sum[o<<1|1]=(add1[o]*sum[o<<1|1]+add[o]*(len>>1))%p;
      
      add[o]=0;
      add1[o]=1;
}
void quary1(LL x,LL y,LL num,LL l,LL r,LL o)
{
     if(x<=l&&y>=r)	
     {
        sum[o]=(sum[o]+(r-l+1)*num)%p;
        add[o]=(add[o]+num)%p;
        return;
     }
    pushdown(r-l+1,o);
    LL mid=(l+r)>>1;
    if(x<=mid)
        quary1(x,y,num,l,mid,o<<1);
    if(y>mid)
        quary1(x,y,num,mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%p;
    return;
}
void quary2(LL x,LL y,LL num,LL l,LL r,LL o)
{
    if(x<=l&&y>=r)
    {
        sum[o]=(sum[o]*num)%p;
        add1[o]=(add1[o]*num)%p;
        add[o]=(add[o]*num)%p;
        return;
    }
    pushdown(r-l+1,o);
    LL mid=(l+r)>>1;
    if(x<=mid)
        quary2(x,y,num,l,mid,o<<1);
    if(y>mid)
        quary2(x,y,num,mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%p;
    return;
}
LL solve(LL x,LL y,LL l,LL r,LL o)
{
    if(x<=l&&y>=r)
        return sum[o];
    pushdown(r-l+1,o);
    LL mid=(l+r)>>1;
    LL res=0;
    if(x<=mid)
        res=(res+solve(x,y,l,mid,o<<1))%p;
    if(y>mid)
        res=(res+solve(x,y,mid+1,r,o<<1|1))%p;
    return res;
}
int main()
{
    LL n,m;
    scanf("%lld%lld%lld",&n,&m,&p);
    build(1,n,1);/*建树操作*/
    while(m--)
    {
        LL op,x,y,k;
        scanf("%lld",&op);
        if(op==1)
        {
            scanf("%lld%lld%lld",&x,&y,&k);
            quary2(x,y,k,1,n,1);/*[x,y]内的每个数乘上k*/
        }
        if(op==2)
        {
            scanf("%lld%lld%lld",&x,&y,&k);
            quary1(x,y,k,1,n,1);/*[x,y]内的每个数加上k*/
        }
        if(op==3)
        {
            scanf("%lld%lld",&x,&y);
            printf("%lld\n",solve(x,y,1,n,1));
        }
    }
    return 0;
}
发布了165 篇原创文章 · 获赞 6 · 访问量 5058

猜你喜欢

转载自blog.csdn.net/lylzsx20172018/article/details/103412969