[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=0;node[num].sum=0;
        return;
    }
    node[num].l=l;node[num].r=r;node[num].add=0;node[num].sum=0;
    build(l,(l+r)/2,num*2);
    build((l+r)/2+1,r,num*2+1);
}
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)/2;
        if(r<=M)add(o*2,l,r,v);
        else if(l>M)add(o*2+1,l,r,v);
        else{
            add(o*2,l,M,v);
            add(o*2+1,M+1,r,v);
        }
    }
    //维护节点o
    if(node[o].l!=node[o].r){//如果区间只是一个元素就不算
        node[o].sum=node[2*o].sum+node[2*o+1].sum;
    }else node[o].sum=0;
    node[o].sum+=node[o].add*(node[o].r-node[o].l+1);
}

//这里addadd是从上往下这条路的累计addadd值[一同回溯记录这条路节点所有add之和，减少了一次回溯累加add值]
//初始时直接令其为0
int sum=0;
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+1));
    }else{
        int M=node[o].l+(node[o].r-node[o].l)/2;
        if(r<=M)ask(o*2,l,r,node[o].add+addadd);
        else if(l>M)ask(o*2+1,l,r,node[o].add+addadd);
        else{
            ask(o*2,l,M,node[o].add+addadd);
            ask(o*2+1,M+1,r,node[o].add+addadd);
        }
    }
}
int main(){
    int T;
    cin>>T;
    int kases=1;
    int i,j;
    int a;
    for(;kases<=T;kases++){
        int N;
        cin>>N;
        build(1,1<<getRight(N),1);
        for(i=1;i<=N;i++){
            cin>>a;
            add(1,i,i,a);
        }
        char str[10];
        cout<<"Case "<<kases<<":\n";
        bool ok=1;
        while(ok){
            cin>>str;
            switch(str[0]){
            case 'Q':
                cin>>i>>j;
                sum=0;
                ask(1,i,j,0);
                cout<<sum<<'\n';
                break;
            case 'A':
                cin>>i>>j;
                add(1,i,i,j);
                break;
            case 'S':
                cin>>i>>j;
                add(1,i,i,-j);
                break;
            case 'E':
                ok=0;
                break;
            default:break;
            }
        }
    }return 0;
}