[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;

}

 

你可能感兴趣的:(数据结构)