线段树系列-pku-3468-A Simple Problem with Integers-区间修改区间求和

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct seg {
int l,r,fa;
long long sum,more;
}tree[401111];
void build(int l,int r,int k) {
tree[k].l=l;
tree[k].r=r;
tree[k].fa=0;           //FA==1表示有该线段的加值
tree[k].more=0;         //more表示该线段的附加值
if(l==r) {
tree[k].sum=a[l];
tree[k].fa=1;
return;
}
int mid = (l+r) >> 1;
build(l,mid,2*k);
build(mid+1,r,2*k+1);
tree[k].sum=tree[2*k].sum+tree[2*k+1].sum;  //除附加值以外的和
}
void update(int l,int r,long long  v,int k) {
if(tree[k].l==l&&tree[k].r==r) {        //如果就是所加线段，直接赋值上附加值
tree[k].more+=v;
tree[k].fa=1;                       //并且标记
return;
}
int mid =(tree[k].l+tree[k].r) >> 1;
if (r<=mid) update(l,r,v,2*k);
else if(l>mid) update(l,r,v,2*k+1);
else {
update(l,mid,v,2*k);
update(mid+1,r,v,2*k+1);
}
tree[k].sum+=v*(r-l+1);             //因为不是对附加值进行操作，所以对sum进行加值
}
long long query(int l,int r,int k) {
long long ans=0;
if(tree[k].fa==1) {                 //统计是否有对应线段的附加值*点数
ans+=tree[k].more*(r-l+1);
}
if(tree[k].l==l&&tree[k].r==r) {
ans+= tree[k].sum;
return ans;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid) return ans+query(l,r,2*k);       //
else if(l>mid) return ans+query(l,r,2*k+1);
else {
return ans+query(l,mid,2*k)+query(mid+1,r,2*k+1);
}
}
int main()
{
int n,q;
char s[10];
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
build(1,n,1);
while(q--) {
scanf("%s",s);
if(s[0]=='Q') {
long long tot;
int x,y;
scanf("%d%d",&x,&y);
tot=query(x,y,1);
printf("%lld\n",tot);
}
else {
int x,y;
long long v;
scanf("%d%d%lld",&x,&y,&v);
update(x,y,v,1);
}
}
return 0;
}