區間更新,區間求和。C的時候在區間[a,b]之間,每個元素增加一個c。Q的時候求出[a,b]區間元素的和。
思路:線段樹。增加一個增量屬性add,更新到范圍內的時候刷新add和value的值,這個時候不再向下傳遞,如果需要向下更新或者向下詢問的時候,更新節點兩個子樹的add和value屬性,這樣需要用的到時候再更新會提高效率。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=100003;
long long sum;
inline int L(int r){return r<<1;}
inline int R(int r){return (r<<1)+1;}
inline int MID(int l,int r){return (l+r)>>1;}
typedef struct
{
int left,right;
long long value,add;
}node;
node tree[MAXN*4];
long long arr[MAXN];
void Update(int);
void Create(int l,int r,int root)
{
tree[root].left=l;
tree[root].right=r;
tree[root].add=0;
if(l==r) {tree[root].value=arr[l];return;}
int mid =MID(l,r);
Create(l,mid,L(root));
Create(mid+1,r,R(root));
tree[root].value=tree[L(root)].value+tree[R(root)].value;
}
void Add(int l,int r,long long v,int root)
{
if(l<=tree[root].left&&tree[root].right<=r)
{
tree[root].add+=v;
tree[root].value+=v*(tree[root].right-tree[root].left+1);
return;
}
Update(root);
if(tree[root].left==tree[root].right) {return;}
int mid=MID(tree[root].left,tree[root].right);
if(l>mid) Add(l,r,v,R(root));
else if(r<=mid) Add(l,r,v,L(root));
else
{
Add(l,mid,v,L(root));
Add(mid+1,r,v,R(root));
}
tree[root].value=tree[L(root)].value+tree[R(root)].value;
}
void Update(int node)//更新節點,把大區間的增值傳給小區間,給小區間的值加上增量
{
if(tree[node].add)
{
tree[L(node)].add+=tree[node].add;//更新子樹時會用到
tree[R(node)].add+=tree[node].add;
tree[L(node)].value+=(tree[L(node)].right-tree[L(node)].left+1)*tree[node].add;//這時候的值就是區間和
tree[R(node)].value+=(tree[R(node)].right-tree[R(node)].left+1)*tree[node].add;
tree[node].add=0;
}
}
void Solve(int l,int r,int root)
{
if(l<=tree[root].left&&tree[root].right<=r)
{
sum+=tree[root].value;
return;
}
Update(root);
if(tree[root].left==tree[root].right) return;
int mid=MID(tree[root].left,tree[root].right);
if(l>mid) Solve(l,r,R(root));
else if(r<=mid) Solve(l,r,L(root));
else
{
Solve(l,mid,L(root));
Solve(mid+1,r,R(root));
}
}
int main()
{
//freopen("input","r",stdin);
int m,n;
while(scanf("%d %d",&m,&n)!=EOF)
{
for(int i=1;i<=m;i++)
{
scanf("%lld",arr+i);
}
Create(1,m,1);
char c[2];
while(n--)
{
scanf("%s",c);
if('C'==c[0])
{
int l,f;
long long v;
scanf("%d %d %lld",&l,&f,&v);
Add(l,f,v,1);
}
else
{
int l,f;
scanf("%d %d",&l,&f);
sum=0;
Solve(l,f,1);
printf("%lld\n",sum);
}
}
}
return 0;
}