例题
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define ll long long
#define MT(a,b) memset(a,b,sizeof(a))
const int maxn=1E5+5;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
int num[maxn];
struct node
{
ll sum,lazy;
}qwe[maxn<<2];//用等比数列的知识可得空间趋近于4N
void update(int root)
{
qwe[root].sum=qwe[root<<1].sum+qwe[root<<1|1].sum;//每个结点的sum=左儿子+右儿子
}
void build_tree(int l,int r,int root)//建立线段树
{
qwe[root].lazy=0;//初始化lazy数组标记
if (l==r){
qwe[root].sum=num[l];//如果l=r说明此时区间只有一个数,是叶子结点
return;
}
int mid=(l+r)>>1;
build_tree(l,mid,root<<1);//这是左儿子
build_tree(mid+1,r,root<<1|1);//这是右儿子
update(root);//更新这个结点的sum
}
void push_down(int root,int l,int r)//将一个结点的lazy赋给它的儿子们,并且将这个lazy加到它的值上
{
if (qwe[root].lazy==0)//如果lazy=0,说明不需要处理
return ;
int mid=(l+r)>>1;//可能有多次lazy的改变所以选择+=
qwe[root<<1].lazy+=qwe[root].lazy;//左结点
qwe[root<<1|1].lazy+=qwe[root].lazy;//右结点
qwe[root<<1].sum+=qwe[root].lazy*(mid-l+1);
qwe[root<<1|1].sum+=qwe[root].lazy*(r-mid);
qwe[root].lazy=0;//初始化这个结点的lazy,因为以及给了儿子并且这个结点的sum已经被lazy改变了
}
void change(int l,int r,int ql,int qr,int root,int val)//让[ql,qr]所有元素加上val
{
if (l==ql&&r==qr){
qwe[root].sum+=val*(r-l+1);//如果已经是目标区间了,处理sum
qwe[root].lazy+=val;//标记上lazy,这样他的子子孙孙也相当于有了保险
return;
}
push_down(root,l,r);//处理lazy标记
int mid=(l+r)>>1;
if (qr<=mid) change(l,mid,ql,qr,root<<1,val);//如果目标区间在这个结点的左侧
else if (ql>mid) change(mid+1,r,ql,qr,root<<1|1,val);//如果目标区间在这个结点的右侧
else{//如果目标区间既有结点左侧,也存在于结点右侧
change(l,mid,ql,mid,root<<1,val);//那么相当于在左儿子找[ql,(l+r)/2]的值+在右儿子找[(l+r)/2+1,qr]的值
change(mid+1,r,mid+1,qr,root<<1|1,val);
}
update(root);
}
ll query (int l,int r,int ql,int qr,int root)//查询区间[ql,qr]的总和
{
if (ql==l&&qr==r)
return qwe[root].sum;
push_down(root,l,r);
int mid=(l+r)>>1;
if (qr<=mid) return query(l,mid,ql,qr,root<<1);
else if (ql>mid) return query(mid+1,r,ql,qr,root<<1|1);
else return query(l,mid,ql,mid,root<<1)+query(mid+1,r,mid+1,qr,root<<1|1);
}
int main ()
{
char s[2];
ll tmp,a,b,c;
int n,q;while (~scanf ("%d%d",&n,&q)){
for (int i=1;i<=n;i++){
scanf ("%lld",&tmp);
num[i]=tmp;
}
build_tree(1,n,1);
for (int i=1;i<=q;i++){
scanf("%s",s);
if (s[0]=='C'){
scanf ("%lld%lld%lld",&a,&b,&c);
change(1,n,a,b,1,c);
} else if (s[0]=='Q'){
scanf ("%lld%lld",&a,&b);
tmp=query(1,n,a,b,1);
printf("%lld\n",tmp);
}
}
// for (int i=1;i<=n;i++)
// printf("%d ",query(1,n,i,i,1));
}
return 0;
}