BZOJ1798: [Ahoi2009]Seq 维护序列seq[线段树]

1798: [Ahoi2009]Seq 维护序列seq

Time Limit: 30 Sec  Memory Limit: 64 MB
Submit: 5504  Solved: 1937
[Submit][Status][Discuss]

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

Source

Day1

---

 

乘法和加法

先乘后加，标记加法结合律随便算一下就好了

一直WA，随便改几个long long就A掉了.....

 

//
//  main.cpp
//  ahoi2009
//
//  Created by Candy on 10/11/16.
//  Copyright © 2016 Candy. All rights reserved.
//

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e5+10;
typedef long long ll;
#define m (l+r)/2
#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
#define lc o<<1
#define rc o<<1|1
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,p,a[N],Q,op,l,r,x;
struct node{
    ll sum;
    ll add,mul;
    node():add(0),mul(1){}
}t[N<<2];
inline void build(int o,int l,int r){
    if(l==r) t[o].sum=a[l]%p;
    else{
        build(lson);
        build(rson);
        t[o].sum=(t[lc].sum+t[rc].sum)%p;
    }
}
inline void paint(int o,int l,int r,ll d,ll v){
    if(v!=1){
        t[o].mul=(t[o].mul*v)%p;
        t[o].add=(t[o].add*v)%p;
        t[o].sum=(t[o].sum*v)%p;
    }
    if(d!=0){
        t[o].add=(t[o].add+d)%p;
        t[o].sum=(t[o].sum+(r-l+1)%p*d)%p;
    }
}
inline void pushDown(int o,int l,int r){
    paint(lson,t[o].add,t[o].mul);
    paint(rson,t[o].add,t[o].mul);
    t[o].add=0;t[o].mul=1;
}
inline void merge(int o){
    t[o].sum=(t[lc].sum+t[rc].sum)%p;
}
inline void mult(int o,int l,int r,int ql,int qr,int v){
    //printf("mult %d %d %d\n",o,l,r);
    if(ql<=l&&r<=qr) paint(o,l,r,0,v);
    else{
        pushDown(o,l,r);
        if(ql<=m) mult(lson,ql,qr,v);
        if(m<qr) mult(rson,ql,qr,v);
        merge(o);
    }
}
inline void add(int o,int l,int r,int ql,int qr,int d){
    if(ql<=l&&r<=qr) paint(o,l,r,d,1);
    else{
        pushDown(o,l,r);
        if(ql<=m) add(lson,ql,qr,d);
        if(m<qr) add(rson,ql,qr,d);
        merge(o);
    }
}
inline ll query(int o,int l,int r,int ql,int qr){
    if(ql<=l&&r<=qr) return t[o].sum;
    else{
        pushDown(o,l,r);
        ll ans=0;
        if(ql<=m) ans+=query(lson,ql,qr);
        if(m<qr) ans+=query(rson,ql,qr);//printf("q %d %d %d %d\n",o,l,r,ans);
        return ans%p;
    }
}
int main(int argc, const char * argv[]){
    //freopen("seqb.in","r",stdin);
    //freopen("seqb.out","w",stdout);
    
    n=read();p=read();
    for(int i=1;i<=n;i++) a[i]=read();
    build(1,1,n);
    Q=read();
    for(int i=1;i<=Q;i++){
        op=read();l=read();r=read();
        if(op==1){
            x=read()%p;mult(1,1,n,l,r,x);
        }else{
            if(op==2){
                x=read()%p;add(1,1,n,l,r,x);
            }else
                printf("%lld\n",query(1,1,n,l,r));
        }
    }
    return 0;
}