数据结构-线段树(c++代码实现)

最新推荐文章于 2024-04-28 14:00:13 发布

随风而散y

最新推荐文章于 2024-04-28 14:00:13 发布

阅读量400

点赞数 11

分类专栏：算法文章标签：数据结构算法图论 c++

本文链接：https://blog.csdn.net/zyang654321/article/details/136423110

版权

算法专栏收录该内容

283 篇文章 1 订阅

订阅专栏

本文介绍了如何使用二叉搜索树（AVL树）结构来存储整数，并实现了`build`、`query`和`modify`函数，分别用于构建树、查询指定范围内的元素和更新元素值。主要关注范围查找和动态更新在数据结构中的应用。

摘要由CSDN通过智能技术生成

#include<iostream>
using namespace std;
const int N =100010;
struct Node{
    int l,r;
    int sum;
}tr[N * 4];
int n,m;
int w[N];

void pushup(int u){
    tr[u].sum = tr[u<<1].sum + tr[u<<1|1].sum;
}

void build(int u,int l,int r){
    if(l==r) tr[u] = {l,r,w[r]};
    else {
        tr[u] = {l,r};
        int mid = l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

int query(int u,int l,int r){
    if(tr[u].l>=l && tr[u].r <= r) return tr[u].sum;
    int mid = tr[u].l + tr[u].r >> 1;
    int sum = 0;
    if(l<=mid) sum = query(u<<1,l,r);
    if(r>mid) sum += query(u<<1|1,l,r);
    return sum;
}

void modify(int u,int x,int v){
    if(tr[u].l==tr[u].r) tr[u].sum += v;
    else {
        int mid = tr[u].l + tr[u].r >> 1;
        if(x<=mid) modify(u<<1,x,v);
        else modify(u<<1|1,x,v);
        pushup(u);
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,1,n);
    int k,a,b;
    while(m--){
        scanf("%d%d%d",&k,&a,&b);
        if(k==0) printf("%d\n",query(1,a,b));
        else modify(1,a,b);
    }
}