线段树 模板

本文介绍了一种基于二叉树的数据结构实现,重点讨论了区间更新与查询的高效算法。通过递归构建二叉树,并实现了初始化、查询及更新操作。文章详细解释了这些操作的时间复杂度,并提供了具体的C++代码实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  1. 对于二叉树来说,2n是左子节点,2n+1是右子节点。
  2. 建树复杂度为O(n),更新和查询的复杂度都是O(lgn)。
  3. 要注意对一个区间进行更新的时候不需要一直更新到底,可以提高效率,需要的时候往下带就行。
#define maxn 50000

struct node{
    int l, r, sum;
    int mid(){
        return (l+r)/2;
    }
};

node Tree[maxn*4];
int value[maxn+10];
int flag;

//初始化树,根节点是1
void init_tree(int root, int l, int r){
    Tree[root].l = l;
    Tree[root].r = r;
    if(l == r)
        Tree[root].sum = value[l];
    else{
        init_tree(2*root, l, (l+r)/2);
        init_tree(2*root+1, (l+r)/2 + 1, r);

        Tree[root].sum = Tree[2*root].sum + Tree[2*root+1].sum;
    }
}

//查找和
int query_tree(int root, int l, int r){
    int m = Tree[root].mid();
    if(l == Tree[root].l && r == Tree[root].r)
        return Tree[root].sum;
    else{
        if(l > m)
            return query_tree(2*root+1, l, r);
        else if(r <= m)
            return query_tree(2*root, l, r);
        else
            return query_tree(2*root, l, m) + query_tree(2*root+1, m+1, r);
    }
}

void update_tree(int root, int idx, int v){
    if(Tree[root].l == Tree[root].r)
        Tree[root].sum += flag*v;
    else{
        Tree[root].sum += flag*v;

        if(idx <= Tree[root].mid())
            update_tree(2*root, idx, v);
        else
            update_tree(2*root+1, idx, v);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值