POJ-3468 A Simple Problem with Integers
/**
POJ - 3468 ZKW 线段树 区间更新+区间求和;
*/
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<map>
#include<cstring>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
using namespace std;
#define ll long long
using namespace std;
const int maxn = 2e5+7;
int M,n;
ll tag[maxn<<2],sum[maxn<<2];
/**
tag[i]: 以i为根节点的叶子所加部分,考虑为什么加tag操作;
对于未满二叉树的点(单个的叶子)需要单独的进行操作(类似于块状树);
*/
void update(int l,int r,ll val){
int numl=0,numr=0,num=1;
/**
numl: l自顶向上遍历的叶子数;
numl: r自顶向上遍历的叶子数;
num: 当前层所包含节点
*/
for(l = l+M-1,r = r+M+1; l^r^1 ;num <<= 1,l >>= 1,r >>= 1){
sum[l] += 1ll*numl*val;//l遍历加上下面叶子的和
sum[r] += 1ll*numr*val;//r遍历加上下面叶子的和
if(l&1^1) tag[l^1] += val,sum[l^1] += val*num, numl += num;//更新以当前节点为根节点的一个块状树所包含的区间+val;
if(r&1) tag[r^1] += val,sum[r^1] += val*num, numr += num;
}
for(;l;l>>=1,r>>=1) sum[l]+=1ll*numl*val,sum[r]+=1ll*numr*val;//上传至1节点;
}
ll query(int l,int r){
int numl=0,numr=0,num=1;
ll ans=0;
for(l = l+M-1, r = r+M+1;l^r^1; l >>= 1,r >>= 1, num <<= 1){
if(tag[l]) ans += 1ll*tag[l]*numl;//当前节点更新的贡献
if(tag[r]) ans += 1ll*tag[r]*numr;
if(l&1^1) ans += sum[l^1],numl += num;
if(r&1) ans += sum[r^1],numr += num;
}
for(;l; l >>= 1,r >>= 1){
ans += 1ll*tag[l]*numl;//一直更新值1的根节点,numl,numr显然是不变的;
ans += 1ll*tag[r]*numr;
}
return ans;
}
char op[20];
int main (){
int q;scanf("%d %d",&n,&q);
for(M=1;M<n;M<<=1);memset(sum,0,sizeof(sum));memset(tag,0,sizeof(tag));
for(int i=M+1;i<=M+n;i++) scanf("%lld",&sum[i]);
for(int i=M-1;i>=1;i--) sum[i] = sum[i<<1]+sum[i<<1|1];
while(q--){
scanf("%s",op);
int l,r;scanf("%d %d",&l,&r);
if(op[0]=='Q') printf("%lld\n",query(l,r));
else {
ll val;scanf("%lld",&val);
update(l,r,val);
}
}
return 0;
}
/**
10 10
1 1 1 1 1 4 1 1 5
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
Q 1 9
C 1 6 300
Q 3 9
C 3 8 2
Q 1 4
*/