线段树 [AHOI2011维护序列]

题意:对一串序列有如下操作。
1.将某个区间的每个数都加上某个数
2.将某个区间的每个数乘上某个数
3.询问区间和
做法:线段树,维护时记录三个数组,一起下传。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long LL;
const int maxn=100010;
LL mod;
int n,p,sak[maxn];
LL sum[maxn<<2];
LL add[maxn<<2],mul[maxn<<2];
int ql,qr;
void push_up(int o){
    sum[o]=(sum[o<<1]+sum[o<<1|1])%mod;
}
void create_tree(int o,int l,int r){
    mul[o]=1;sum[o]=0;
    if(l==r){
        scanf("%lld",&sum[o]);
        sum[o]=sum[o]%mod;
        return;
    }
    int mid=(l+r)>>1;
    create_tree(o<<1,l,mid);
    create_tree(o<<1|1,mid+1,r);
    push_up(o);
}
void Mod(int o){
    sum[o]%=mod;
    add[o]%=mod;
    mul[o]%=mod;
}
void push_down(int o,int l,int r,int mid){
    int u=o<<1,v=o<<1|1;
    add[u]=add[u]*mul[o]+add[o];
    add[v]=add[v]*mul[o]+add[o];
    mul[u]=mul[o]*mul[u];
    mul[v]=mul[o]*mul[v];
    sum[u]=sum[u]*mul[o]+add[o]*(mid-l+1);
    sum[v]=sum[v]*mul[o]+add[o]*(r-mid);
    mul[o]=1;add[o]=0;
    Mod(u);Mod(v);
}
void update_mul(int o,int l,int r,int t){
    if(ql<=l && qr>=r){
        mul[o]*=t;
        sum[o]*=t;
        add[o]*=t;
        Mod(o);
        return;
    }
    int mid=(l+r)>>1;
    push_down(o,l,r,mid);
    Mod(o<<1),Mod(o<<1|1);
    if(ql<=mid)update_mul(o<<1,l,mid,t);
    if(qr>mid)update_mul(o<<1|1,mid+1,r,t);
    push_up(o);
}
void update_add(int o,int l,int r,int t){
    if(ql<=l && qr>=r){
        sum[o]+=t*(r-l+1);
        add[o]+=t;
        Mod(o);
        return;
    }
    int mid=(l+r)>>1;
    push_down(o,l,r,mid);
    Mod(o<<1);Mod(o<<1|1);
    if(ql<=mid)update_add(o<<1,l,mid,t);
    if(qr>mid)update_add(o<<1|1,mid+1,r,t);
    push_up(o);
}
int query(int o,int l,int r){
    if(ql<=l && qr>=r){
        return sum[o];
    }
    int mid=(l+r)>>1;
    push_down(o,l,r,mid);
    Mod(o<<1);Mod(o<<1|1);
    int ret=0;
    if(ql<=mid) ret+=query(o<<1,l,mid);
    if(qr>mid) ret+=query(o<<1|1,mid+1,r);
    push_up(o);
    return ret%mod;
}
int main(){
    scanf("%d%lld",&n,&mod);
    create_tree(1,1,n);
    int q;
    scanf("%d",&q);
    while(q--){
        int op,t;
        scanf("%d%d%d",&op,&ql,&qr);
        if(op==1){
            scanf("%d",&t);
            update_mul(1,1,n,t);
        }else if(op==2){
            scanf("%d",&t);
            update_add(1,1,n,t);
        }else{
            printf("%d\n",query(1,1,n));
        }
    }
    return 0;
}

^_^

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值