BZOJ1798 维护序列seq

题目

Description

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为N的数列，不妨设为a1,a2,…,aN 。有如下三种操作形式： (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。

Input

第一行两个整数N和P(1≤P≤1000000000）。第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。第三行有一个整数M，表示操作总数。从第四行开始每行描述一个操作，输入的操作有以下三种形式： 操作1：“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c (1≤t≤g≤N,0≤c≤1000000000)。 操作2：“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。 操作3：“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

Output

对每个操作3，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

Sample Input

7 43

1 2 3 4 5 6 7

5

1 2 5 5

3 2 4

2 3 7 9

3 1 3

3 4 7

Sample Output

2

35

8

HINT

【样例说明】

初始时数列为(1,2,3,4,5,6,7)。
经过第1次操作后，数列为(1,10,15,20,25,6,7)。
对第2次操作，和为10+15+20=45，模43的结果是2。
经过第3次操作后，数列为(1,10,24,29,34,15,16}
对第4次操作，和为1+10+24=35，模43的结果是35。
对第5次操作，和为29+34+15+16=94,模43的结果是8。

测试数据规模如下表所示

数据编号
1 2 3 4 5 6 7 8 9 10
N= 10 1000 1000 10000 60000 70000 80000 90000 100000 100000
M= 10 1000 1000 10000 60000 70000 80000 90000 100000 100000

代码

线段树记录区间和，记录加法和乘法的标记。
下传时注意儿子节点要乘上父节点的乘法标记，儿子节点的加法标记也是一样。
用1和0使加法和乘法操作共用一个update函数。

#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdio>
#define R0(i,n) for(register int i=0;i<n;++i)
#define R1(i,n) for(register int i=1;i<=n;++i)
#define cl(x,c) memset(x,c,sizeof x)
#define INF 0x7fffffff
#define MAXN 110000
#define lson k<<1,l,m
#define rson k<<1|1,m+1,r
#define lc k<<1
#define rc k<<1|1
using namespace std;
typedef long long ll;
template <class T> inline void read(T&x){
    bool f = false; char ch;
    for (ch = getchar(); ch <= 32; ch = getchar());
    if (ch == ‘-‘) f = true, ch = getchar();
    for (x = 0; ch > 32; ch = getchar()) x = x * 10 + ch - ‘0‘;
    if (f) x=-x;
}
template <class T> inline void write(T x){
    if (x < 0) putchar(‘-‘), x=-x;
    if (x < 10)putchar(x+‘0‘);
    else write(x/10), putchar(x%10+‘0‘);
}
ll sum[MAXN<<2],P,mul[MAXN<<2],add[MAXN<<2];
int n,m;
void push_down(int k,int m){
    sum[lc]=(sum[lc]*mul[k]+(m-(m>>1))*add[k])%P;
    sum[rc]=(sum[rc]*mul[k]+(m>>1)*add[k])%P;
    mul[lc]=mul[lc]*mul[k]%P;
    mul[rc]=mul[rc]*mul[k]%P;
    add[lc]=(add[lc]*mul[k]+add[k])%P;
    add[rc]=(add[rc]*mul[k]+add[k])%P;
    mul[k]=1;add[k]=0;
}
void build(int k,int l,int r){
    mul[k]=1;add[k]=0;
    if(l==r){
        read(sum[k]);
        return;
    }
    int m=(l+r)>>1;
    build(lson),build(rson);
    sum[k]=(sum[lc]+sum[rc])%P;
}
void update(int k,int l,int r,int L,int R,int M,int A){
    if(l>=L&&r<=R){
        sum[k]=(sum[k]*M+(r-l+1)*A)%P;
        mul[k]=mul[k]*M%P;
        add[k]=(add[k]*M+A)%P;
        return;
    }
    int m=(l+r)>>1;
    push_down(k,r-l+1);
    if(L<=m)update(lson,L,R,M,A);
    if(R>m) update(rson,L,R,M,A);
    sum[k]=(sum[lc]+sum[rc])%P;
}
ll query(int k,int l,int r,int L,int R){
    if(l>=L&&r<=R) return sum[k]%P;
    int m=(l+r)>>1;
    push_down(k,r-l+1);
    ll ret=0;
    if(L<=m) ret=(ret+query(lson,L,R))%P;
    if(R>m)  ret=(ret+query(rson,L,R))%P;
    sum[k]=(sum[lc]+sum[rc])%P;
    return ret%P;
}
int main(){
    read(n),read(P);
    build(1,1,n);
    read(m);
    int u,v,w,x;
    R0(i,m){
        read(u);
       if(u==1)read(v),read(w),read(x),update(1,1,n,v,w,x,0);
        else if(u==2)read(v),read(w),read(x),update(1,1,n,v,w,1,x);
        else read(v),read(w),write(query(1,1,n,v,w)),puts("");
        }
    return 0;
}