[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;
}
复制代码
相关文章
|
7月前
|
设计模式 算法 搜索推荐
C++数据结构设计:理解并选择策略模式与模板特化
C++数据结构设计:理解并选择策略模式与模板特化
98 2
|
7月前
|
存储 人工智能 索引
【数据结构】树状数组和线段树
【数据结构】树状数组和线段树
68 0
ACM刷题之路(十三)数据结构——链表
ACM刷题之路(十三)数据结构——链表
109 0
|
7月前
|
算法 索引 Python
Python高级数据结构——线段树(Segment Tree)
Python高级数据结构——线段树(Segment Tree)
294 2
|
存储 人工智能
高级数据结构-线段树
线段树树基于分治思想的二叉树,用来维护区间信息(区间和、区间最大值、区间最小值等等)。可以在 O(logn) 的时间内完成区间信息的查询和修改。
72 0
高级数据结构-线段树
|
C语言 C++
ACM刷题之路(十二)数据结构——顺序表
ACM刷题之路(十二)数据结构——顺序表
|
算法 容器
【牛客题霸】【模板】栈|栈的压入、弹出序列(下)
【牛客题霸】【模板】栈|栈的压入、弹出序列
81 0
|
存储 算法 C++
单调栈模板总结及应用
单调栈模板总结及应用
96 0
数据结构:线段树代码详解
数据结构:线段树代码详解
92 0
数据结构:线段树代码详解