线段树 模板

  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);
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值