洛谷P3373 【模板】线段树 2

题目描述

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

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）

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,p,judge,x,y,k;
long long a[100007];
long long read()
{
    long long x=0,f=1;
    char ch=getchar();
    while(ch>'9'||ch<'0')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
struct node
{
    long long v,mul,add;
} st[400007];
void build(int root, int l, int r)
{
    st[root].mul=1;
    st[root].add=0;
    if(l==r)
        st[root].v=a[l];
    else
    {
        int m=(l+r)/2;
        build(root*2, l, m);
        build(root*2+1, m+1, r);
        st[root].v=st[root*2].v+st[root*2+1].v;
    }
    st[root].v%=p;
    return ;
}
void pushdown(int root, int l, int r)
{
    int m=(l+r)/2;
    st[root*2].v=(st[root*2].v*st[root].mul+st[root].add*(m-l+1))%p;
    st[root*2+1].v=(st[root*2+1].v*st[root].mul+st[root].add*(r-m))%p;
    st[root*2].mul=(st[root*2].mul*st[root].mul)%p;
    st[root*2+1].mul=(st[root*2+1].mul*st[root].mul)%p;
    st[root*2].add=(st[root*2].add*st[root].mul+st[root].add)%p;
    st[root*2+1].add=(st[root*2+1].add*st[root].mul+st[root].add)%p;
    st[root].mul=1;
    st[root].add=0;
    return ;
}
void up(int root,int stdl,int stdr,int l,int r,long long k)
{
    if(r<stdl||stdr<l)
        return ;
    if(l<=stdl&&stdr<=r)
    {
        st[root].v=(st[root].v*k)%p;
        st[root].mul=(st[root].mul*k)%p;
        st[root].add=(st[root].add*k)%p;
        return ;
    }
    pushdown(root, stdl, stdr);
    int m=(stdl+stdr)/2;
    up(root*2,stdl,m,l,r,k);
    up(root*2+1,m+1,stdr,l,r,k);
    st[root].v=(st[root*2].v+st[root*2+1].v)%p;
    return ;
}
void date(int root,int stdl,int stdr,int l,int r,long long k)
{
    if(r<stdl||stdr<l)
        return ;
    if(l<=stdl&&stdr<=r)
    {
        st[root].add=(st[root].add+k)%p;
        st[root].v=(st[root].v+k*(stdr-stdl+1))%p;
        return ;
    }
    pushdown(root,stdl,stdr);
    int m=(stdl+stdr)/2;
    date(root*2,stdl,m,l,r,k);
    date(root*2+1,m+1,stdr,l,r,k);
    st[root].v=(st[root*2].v+st[root*2+1].v)%p;
    return ;
}
long long query(int root,int stdl,int stdr,int l,int r)
{
    if(r<stdl||stdr<l)
        return 0;
    if(l<=stdl&&stdr<=r)
        return st[root].v;
    pushdown(root,stdl,stdr);
    int m=(stdl+stdr)/2;
    return (query(root*2,stdl,m,l,r)+query(root*2+1,m+1,stdr,l,r))%p;
}
int main()
{
    n=read(),m=read(),p=read();
    for(int i=1; i<=n; i++)
        a[i]=read();
    build(1,1,n);
    while(m--)
    {
        judge=read();
        if(judge==1)
        {
            x=read(),y=read(),k=read();
            up(1,1,n,x,y,k);
        }
        else if(judge==2)
        {
            x=read(),y=read(),k=read();
            date(1,1,n,x,y,k);
        }
        else
        {
            x=read(),y=read();
            printf("%lld
", query(1,1,n,x,y));
        }
    }
    return 0;
}