题意:
给出l,r,x,对l到r上的每个数都加上x;
给出l,r,对l到r求和
思路:线段树延迟更新
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string.h>
using namespace std;
#define maxn 2000005
typedef long long ll;
ll da[maxn*2+5],db[maxn*2+5];
int n,q;
void add(int a,int b,int x,int k,int l,int r)
{
if(a<=l&&r<=b){
da[k]+=x;
}
else if(b>l&&a<r){
db[k]+=(min(r,b)-max(l,a))*x;
add(a,b,x,2*k+1,l,(l+r)/2);
add(a,b,x,2*k+2,(l+r)/2,r);//左闭右开
}
}
ll sum(int a,int b,int k,int l,int r)
{
if(l>=b||r<=a) return 0;
else if(a<=l&&r<=b){
return da[k]*(r-l)+db[k];
}
else{
ll res=(min(b,r)-max(l,a))*da[k];//应该新加的值
res+=sum(a,b,2*k+1,l,(l+r)/2);
res+=sum(a,b,2*k+2,(l+r)/2,r);
return res;
}
}
int main()
{
while(~scanf("%d%d",&n,&q)){
memset(da,0,sizeof(da));
memset(db,0,sizeof(db));
for(int i=0;i<n;i++){
int x;
scanf("%d",&x);
add(i,i+1,x,0,0,n);
}
char str[3];
for(int i=0;i<q;i++){
scanf("%s",str);
int L,R,x;
if(str[0]=='C'){
scanf("%d%d%d",&L,&R,&x); L--;R--;
add(L,R+1,x,0,0,n);
}
else{
scanf("%d%d",&L,&R);L--;R--;
printf("%lld\n",sum(L,R+1,0,0,n));
}
}
}
}