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<bits/stdc++.h>
#define ll long long
#define lson (rt<<1)
#define rson (rt<<1|1)
using namespace std;

const int maxn = 1e5+10;

inline int read () {
    int x=0;    char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}
    return x;
}

struct node {
    ll val,add,mul;
}tree[maxn<<2];

ll a[maxn],p;

int n,m;

void pushup(int rt) {
    tree[rt].val=(tree[lson].val+tree[rson].val)%p;
}

void pushdown(int l,int r,int rt) {
    int mid = (l+r)>>1;
    tree[lson].val=(tree[lson].val*tree[rt].mul+tree[rt].add*(mid-l+1))%p;
    tree[rson].val=(tree[rson].val*tree[rt].mul+tree[rt].add*(r-mid))%p;
    tree[lson].mul=(tree[lson].mul*tree[rt].mul)%p;
    tree[rson].mul=(tree[rson].mul*tree[rt].mul)%p;
    tree[lson].add=(tree[lson].add*tree[rt].mul+tree[rt].add)%p;
    tree[rson].add=(tree[rson].add*tree[rt].mul+tree[rt].add)%p;
    tree[rt].mul=1;
    tree[rt].add=0;
}

void buildtree(int l,int r,int rt) {
    tree[rt].mul=1;
    if(l==r) {
        tree[rt].val=a[l]%p;
    }
    else {
           int mid = (l+r)>>1;
        if(mid>=l) buildtree(l,mid,lson);
        if(mid+1<=r) buildtree(mid+1,r,rson);    
        pushup(rt);     
    }

}

void updata1(int x,int y,ll delta,int l,int r,int rt) {
    if(x<=l&&y>=r) {
        tree[rt].val=(tree[rt].val*delta)%p;
        tree[rt].mul=(tree[rt].mul*delta)%p;
        tree[rt].add=(tree[rt].add*delta)%p;
        return ;
    }
    pushdown(l,r,rt);
    int mid = (l+r)>>1;
    if(x<=mid) updata1(x,y,delta,l,mid,lson);
    if(y>mid) updata1(x,y,delta,mid+1,r,rson);
    pushup(rt);
}

void updata2(int x,int y,ll delta,int l,int r,int rt) {
    if(x<=l&&y>=r) {
        tree[rt].val=(tree[rt].val+delta*(r-l+1))%p;
        tree[rt].add+=delta;
        tree[rt].add%=p;
        return ;
    }
    pushdown(l,r,rt);
    int mid = (l+r)>>1;
    if(x<=mid) updata2(x,y,delta,l,mid,lson);
    if(y>mid) updata2(x,y,delta,mid+1,r,rson);
    pushup(rt);
} 

ll query(int x,int y,int l,int r,int rt) {
    if(x<=l&&y>=r) {
        return tree[rt].val;
    }
    pushdown(l,r,rt);
    int mid = (l+r)>>1;
    ll ans = 0;
    if(x<=mid) ans+=query(x,y,l,mid,lson);
    if(y>mid) ans+=query(x,y,mid+1,r,rson);
    return ans%p;
}

int main() {
    n=read();m=read();scanf("%lld",&p);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    buildtree(1,n,1);
    while(m--) {
        int op;scanf("%d",&op);
        if(op==1) {
            int x,y;ll z;
            scanf("%d%d%lld",&x,&y,&z);
            updata1(x,y,z,1,n,1);
        }  
        else if(op==2) {
            int x,y;ll z;
            scanf("%d%d%lld",&x,&y,&z);
            updata2(x,y,z,1,n,1);
        }
        else {
            int x,y;scanf("%d%d",&x,&y);
            printf("%lld
",query(x,y,1,n,1)%p);
        }
    }
    return 0;
}