[省选前题目整理][POJ 3468]A Simple Problem with Integers(线段树区间修改)

题目链接

题目大意

思路

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 401000

using namespace std;

typedef long long int LL;

LL a[MAXN];
LL sum[MAXN],addtag[MAXN];

void pushup(int o)
{
sum[o]=sum[o<<1]+sum[o<<1|1];
}

void pushdown(int o,int L,int R)
{
if(addtag[o])
{
int M=(L+R)>>1;
addtag[o<<1]+=addtag[o];
addtag[o<<1|1]+=addtag[o];
sum[o<<1]+=(M-L+1)*addtag[o];
sum[o<<1|1]+=(R-M)*addtag[o];
addtag[o]=0;
}
}

void build(int o,int L,int R)
{
if(L==R)
{
sum[o]=a[L];
addtag[o]=0;
return;
}
LL M=(L+R)>>1;
build(o<<1,L,M);
build(o<<1|1,M+1,R);
pushup(o);
addtag[o]=0;
}

void update(int o,int L,int R,int ql,int qr,LL addv)
{
if(ql<=L&&R<=qr)
{
addtag[o]+=addv;
sum[o]+=(R-L+1)*addv; //!!!!!
return;
}
pushdown(o,L,R);
int M=(L+R)>>1;
if(ql<=M) update(o<<1,L,M,ql,qr,addv);
if(qr>M) update(o<<1|1,M+1,R,ql,qr,addv);
pushup(o);
}

LL query(int o,int L,int R,int ql,int qr)
{
if(ql<=L&&R<=qr)
return sum[o];
pushdown(o,L,R);
int M=(L+R)>>1;
LL ans=0;
if(ql<=M) ans+=query(o<<1,L,M,ql,qr);
if(qr>M) ans+=query(o<<1|1,M+1,R,ql,qr);
return ans;
}

int main()
{
int n,q;
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
build(1,1,n);
for(int i=1;i<=q;i++)
{
char cmd[4];
LL ql,qr,val;
scanf("%s%lld%lld",cmd,&ql,&qr);
if(cmd[0]=='C')
{
scanf("%lld",&val);
update(1,1,n,ql,qr,val);
}
else printf("%lld\n",query(1,1,n,ql,qr));
}
return 0;
}