【线段树 区间加 区间乘】 洛谷 P3373 【模板】线段树 2

P3373 【模板】线段树 2 

https://www.luogu.org/problem/P3373

题目描述

如题，已知一个数列，你需要进行下面三种操作：

1.将某区间每一个数乘上x

2.将某区间每一个数加上x

3.求出某区间每一个数的和

输入格式

第一行包含三个整数N、M、P，分别表示该数列数字的个数、操作的总个数和模数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数乘上k

操作2： 格式：2 x y k 含义：将区间[x,y]内每个数加上k

操作3： 格式：3 x y 含义：输出区间[x,y]内每个数的和对P取模所得的结果

输出格式

输出包含若干行整数，即为所有操作3的结果。

输入输出样例

输入 #1复制

5 5 38
1 5 4 2 3
2 1 4 1
3 2 5
1 2 4 2
2 3 5 5
3 1 4

输出 #1复制

17
2

说明/提示

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_^）

样例说明：

故输出应为17、2（40 mod 38=2）

/*
（x + 2）* 10 = x * 10 + 2 * 10
所以我们需要一个lazy标记记录加，一个lazy标记记录*
重点是*对+会有影响，所以add[rt]=add[rt]*k%mod
当你传递时，子结点的add标记也是先*后+，保证正确性
重点就是* + 中为了使答案正确，应该先*再+ 
*/


#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+10;
ll sum[maxn*4],add[maxn*4],mul[maxn*4];
ll mod;

void pushup(int rt)
{
    if(rt)
        sum[rt]=(sum[2*rt]+sum[2*rt+1])%mod;
}

void pushdown(int rt,int len)
{
    sum[2*rt]=sum[2*rt]*mul[rt]%mod;
    sum[2*rt+1]=sum[2*rt+1]*mul[rt]%mod;
    sum[2*rt]=(sum[2*rt]+add[rt]*(len-len/2))%mod;
    sum[2*rt+1]=(sum[2*rt+1]+add[rt]*(len/2))%mod;
    add[2*rt]=(add[2*rt]*mul[rt])%mod,add[2*rt+1]=(add[2*rt+1]*mul[rt])%mod;
    add[2*rt]=(add[2*rt]+add[rt])%mod,add[2*rt+1]=(add[2*rt+1]+add[rt])%mod;
    mul[2*rt]=(mul[2*rt]*mul[rt])%mod,mul[2*rt+1]=(mul[2*rt+1]*mul[rt])%mod;
    add[rt]=0,mul[rt]=1;
}

void build(int l,int r,int rt)
{
    add[rt]=0;
    mul[rt]=1;
    if(l==r)
    {
        scanf("%lld",&sum[rt]);
        sum[rt]%=mod;
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,2*rt);
    build(mid+1,r,2*rt+1);
    pushup(rt);
}

void update(int x,int y,int l,int r,int rt,ll k,int flag) //乘
{
    if(x<=l&&y>=r)
    {
        if(flag==2)
        {
            add[rt]=(add[rt]+k)%mod;
            sum[rt]=(sum[rt]+k*(r-l+1))%mod;
        }
        else
        {
            add[rt]=add[rt]*k%mod;
            mul[rt]=mul[rt]*k%mod;
            sum[rt]=sum[rt]*k%mod;
        }
        return;
    }
    pushdown(rt,r-l+1);
    int mid=(l+r)/2;
    if(x<=mid) update(x,y,l,mid,2*rt,k,flag);
    if(y>mid) update(x,y,mid+1,r,2*rt+1,k,flag);
    pushup(rt);
}



ll query(int x,int y,int l,int r,int rt)
{
    if(x<=l&&y>=r)
    {
        return sum[rt]%mod;
    }
    pushdown(rt,r-l+1);
    int mid=(l+r)/2;
    ll ans=0;
    if(x<=mid) ans=(ans+query(x,y,l,mid,2*rt))%mod;
    if(y>mid) ans=(ans+query(x,y,mid+1,r,2*rt+1))%mod;
    pushup(rt);
    return ans%mod;
}

int main()
{
    int n,m,opt,x,y;
    ll k;
    scanf("%d%d%lld",&n,&m,&mod);
    build(1,n,1);
    while(m--)
    {
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%lld",&x,&y,&k);
            update(x,y,1,n,1,k%mod,1);
        }
        else if(opt==2)
        {
            scanf("%d%d%lld",&x,&y,&k);
            update(x,y,1,n,1,k%mod,2);
        }
        else
        {
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(x,y,1,n,1)%mod);
        }
    }

    return 0;
}
/*

5 100 39
1 2 3 4 5
1 1 5 6
2 1 5 7
1 1 5 7
2 2 5 3
3 1 4

*/