[ACM_数据结构] 线段树模板

#include<iostream>
#include<cmath>
using namespace std;
#define maxn 200005
class Node{
public:
    int l,r;
    int add;//附加值
    int sum;
}node[maxn];
int getRight(int n){//获得满足2^x>=n的最小x[从0层开始,给编号获得层数]
    return ceil(log10(n*1.0)/log10(2.0));
}
void build(int l,int r,int num){//输入区间[1,2^getRight(n)],num=1建树
    if(l==r){
        node[num].l=node[num].r=l;node[num].add=;node[num].sum=;
        return;
    }
    node[num].l=l;node[num].r=r;node[num].add=;node[num].sum=;
    build(l,(l+r)/,num*);
    build((l+r)/+,r,num*+);
}
void add(int o,int l,int r,int v){//从o节点开始递归[只要调用时o=1即可]在区间[l,r]全部加v
    if(l<=node[o].l && r>=node[o].r){//全覆盖[递归边界]
        node[o].add+=v;
    }else{
        int M=node[o].l+(node[o].r-node[o].l)/;
        if(r<=M)add(o*,l,r,v);
        else if(l>M)add(o*+,l,r,v);
        else{
            add(o*,l,M,v);
            add(o*+,M+,r,v);
        }
    }
    //维护节点o
    if(node[o].l!=node[o].r){//如果区间只是一个元素就不算
        node[o].sum=node[*o].sum+node[*o+].sum;
    }else node[o].sum=;
    node[o].sum+=node[o].add*(node[o].r-node[o].l+);
}

//这里addadd是从上往下这条路的累计addadd值[一同回溯记录这条路节点所有add之和，减少了一次回溯累加add值]
//初始时直接令其为0
int sum=;
void ask(int o,int l,int r,int addadd){//从o节点开始递归[只要调用时o=1即可]在区间[l,r]的和
    if(l<=node[o].l && r>=node[o].r){//全覆盖[递归边界]
        sum+=(node[o].sum+addadd*(node[o].r-node[o].l+));
    }else{
        int M=node[o].l+(node[o].r-node[o].l)/;
        if(r<=M)ask(o*,l,r,node[o].add+addadd);
        else if(l>M)ask(o*+,l,r,node[o].add+addadd);
        else{
            ask(o*,l,M,node[o].add+addadd);
            ask(o*+,M+,r,node[o].add+addadd);
        }
    }
}
int main(){
    int T;
    cin>>T;
    int kases=;
    int i,j;
    int a;
    for(;kases<=T;kases++){
        int N;
        cin>>N;
        build(,<<getRight(N),);
        for(i=;i<=N;i++){
            cin>>a;
            add(,i,i,a);
        }
        char str[];
        cout<<"Case "<<kases<<":\n";
        bool ok=;
        while(ok){
            cin>>str;
            switch(str[]){
            case 'Q':
                cin>>i>>j;
                sum=;
                ask(,i,j,);
                cout<<sum<<'\n';
                break;
            case 'A':
                cin>>i>>j;
                add(,i,i,j);
                break;
            case 'S':
                cin>>i>>j;
                add(,i,i,-j);
                break;
            case 'E':
                ok=;
                break;
            default:break;
            }
        }
    }return ;
}