线段树区间更新模板

最大的遗憾是没写区间乘法的= =



int b[100001<<2];
int lazy[100001<<2];

void pushup(int rt)
{
    b[rt]=b[rt<<1]+b[rt<<1|1];
}

void pushdown(int rt,int len)
{
    if(lazy[rt])
    {
        lazy[rt<<1] += lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        b[rt<<1] += lazy[rt]*(len-(len>>1));
        b[rt<<1|1]+=lazy[rt]*(len>>1);
        lazy[rt] = 0;
    }
}

void build(int l,int r,int rt)
{
    lazy[rt] = 0;
    if(l==r)
    {
        scanf("%d",&b[rt]);
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    pushup(rt);
}

void update(int L,int R,int c,int l,int r,int rt)
{
    if(l>=L&&R>=r)
    {
        lazy[rt]+=c;
        b[rt]+=c*(r-l+1);
        return ;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(m>=L) update(L,R,c,l,m,rt<<1);
    if(m<R) update(L,R,c,m+1,r,rt<<1|1);
    pushup(rt);
}

long long int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r) return b[rt];
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    long long int ret=0;
    if(m>=L) ret+=query(L,R,l,m,rt<<1);
    if(m<R) ret+=query(L,R,m+1,r,rt<<1|1);
    return ret;
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        int i;
        build(1,n,1);
        int m;
        scanf("%d",&m);
        for(i=0;i<m;i++)
        {
            char ch[5];
            scanf("%s",ch);
            int a,b,c;
            if(ch[0]=='A')
            {
                scanf("%d%d%d",&a,&b,&c);
                update(a,b,c,1,n,1);
            }
            else if(ch[0]=='Q')
            {
                scanf("%d%d",&a,&b);
                printf("%lld\n",query(a,b,1,n,1));
            }
        }
    }
    return 0;
}





#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }
 
const int N=100005;
#define lc x<<1
#define rc x<<1|1
#define MID (l+r)>>1
#define lson l, mid, lc
#define rson mid+1, r, rc
 
int n, MD;
 
struct node {
    int sum, add, mul;
    void upd(int a, int m, int len) {
        add=((ll)add*m+a)%MD;
        mul=((ll)mul*m)%MD;
        sum=((ll)sum*m+(ll)a*len)%MD;
    }
}t[N<<2];
void pushdown(int x, int len) {
    if(t[x].add!=0 || t[x].mul!=1)
        t[lc].upd(t[x].add, t[x].mul, (len-(len>>1))),
        t[rc].upd(t[x].add, t[x].mul, len>>1),
        t[x].add=0, t[x].mul=1;
}
void pushup(int x) { t[x].sum=(t[lc].sum+t[rc].sum)%MD; }
void build(int l, int r, int x) {
    t[x].add=0;
    t[x].mul=1;
    if(l==r) { t[x].sum=getint(); return; }
    int mid=MID;
    build(lson); build(rson);
    pushup(x);
}
void update(int l, int r, int x, int L, int R, int add, int mul) {
    if(L<=l && r<=R) { t[x].upd(add, mul, r-l+1); return; }
    pushdown(x, r-l+1);
    int mid=MID;
    if(L<=mid) update(lson, L, R, add, mul);
    if(mid<R) update(rson, L, R, add, mul);
    pushup(x);
}
int query(int l, int r, int x, int L, int R) {
    if(L<=l && r<=R) return t[x].sum;
    pushdown(x, r-l+1);
    int mid=MID, ret=0;
    if(L<=mid) ret+=query(lson, L, R);
    if(mid<R) ret+=query(rson, L, R);
    ret%=MD;
    return ret;
}
int main() {
    read(n); read(MD);
    build(1, n, 1);
    int m=getint();
    while(m--) {
        int c=getint();
        if(c==1) { int l=getint(), r=getint(), x=getint(); update(1, n, 1, l, r, 0, x); }
        else if(c==2) { int l=getint(), r=getint(), x=getint(); update(1, n, 1, l, r, x, 1); }
        else if(c==3) { int l=getint(), r=getint(); printf("%d\n", query(1, n, 1, l, r)); }
    }
    return 0;
}

/wangshangmuban


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值