[BZOJ1798][Ahoi2009]Seq 维护序列seq

最新推荐文章于 2018-09-26 18:58:04 发布

Zvezda_

最新推荐文章于 2018-09-26 18:58:04 发布

阅读量504

点赞数

分类专栏：数据结构_线段树 BZOJ

本文链接：https://blog.csdn.net/Zvezda_/article/details/46960269

版权

BZOJ 同时被 2 个专栏收录

159 篇文章 0 订阅

订阅专栏

数据结构_线段树

13 篇文章 0 订阅

订阅专栏

原题地址

线段树模(mu)板题.

AC code:

#include <cstdio>
typedef long long ll;
const int N=100010;
ll n,q,m,tot;
ll a[N];

struct nod{
    ll  l,r,sum,tmul,tadd;
    nod *lc,*rc;
}pool[N<<3];

struct Segtree{
    nod *root;

    Segtree(){
        build(&root,1,n);
    }

    void build(nod **p,ll L,ll R){
        *p=&pool[tot++];(*p)->l=L;(*p)->r=R;(*p)->tmul=1;
        if(L==R){
            (*p)->sum=a[L]%q;
            return ;
        }
        ll M=(L+R)>>1;
        build(&(*p)->lc,L,M);
        build(&(*p)->rc,M+1,R);
        (*p)->sum=((*p)->lc->sum+(*p)->rc->sum)%q;
    }
    void clear(nod *p){
        if(p->l!=p->r){
            p->lc->tmul=(p->lc->tmul*p->tmul)%q;
            p->lc->tadd=(p->lc->tadd*p->tmul)%q;
            p->rc->tmul=(p->rc->tmul*p->tmul)%q;
            p->rc->tadd=(p->rc->tadd*p->tmul)%q;
            p->lc->tadd=(p->lc->tadd+p->tadd)%q;
            p->rc->tadd=(p->rc->tadd+p->tadd)%q;
        }
        p->sum=(p->sum*p->tmul+(p->r-p->l+1)*p->tadd)%q;
        p->tadd=0;p->tmul=1;
    }
    void mul(nod *p,ll L,ll R,ll v){
        clear(p);
        if(p->l==L&&p->r==R){
            p->tmul=v;
            return ;
        }
        ll M=(p->l+p->r)>>1;
        if(R<=M) mul(p->lc,L,R,v);
        else if(L>M) mul(p->rc,L,R,v);
        else{
            mul(p->lc,L,M,v);
            mul(p->rc,M+1,R,v);
        }
        clear(p->lc);clear(p->rc);
        p->sum=(p->lc->sum+p->rc->sum)%q;
    }
    void add(nod *p,ll L,ll R,ll v){
        clear(p);
        if(p->l==L&&p->r==R){
            p->tadd=v;
            return ;
        }
        ll M=(p->l+p->r)>>1;
        if(R<=M) add(p->lc,L,R,v);
        else if(L>M) add(p->rc,L,R,v);
        else{
            add(p->lc,L,M,v);
            add(p->rc,M+1,R,v);
        }
        clear(p->lc);clear(p->rc);
        p->sum=(p->lc->sum+p->rc->sum)%q;
    }
    ll query(nod *p,ll L,ll R){
        clear(p);
        if(p->l==L&&p->r==R) return p->sum%q;
        ll M=(p->l+p->r)>>1;
        if(R<=M) return query(p->lc,L,R)%q;
        else if(L>M) return query(p->rc,L,R)%q;
        else return (query(p->lc,L,M)+query(p->rc,M+1,R))%q;
    }
};

int main(){
    scanf("%lld%lld",&n,&q);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    Segtree T;
    scanf("%lld",&m);
    for(int i=1;i<=m;i++){
        ll t,L,R,v;
        scanf("%lld%lld%lld",&t,&L,&R);
        if(t!=3){
            scanf("%lld",&v);
            if(t==1) T.mul(T.root,L,R,v);
            else T.add(T.root,L,R,v);
        }
        else printf("%lld\n",T.query(T.root,L,R));
    }

    return 0;
}