POJ - 3468 ZKW 线段树 区间更新+区间求和;

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
*/