树状数组c++详解

树状数组(Binary Indexed Tree,简称BIT)是一种用于高效处理数组范围查询和单点更新操作的数据结构。它解决了传统的线段树在处理范围查询时的低效问题,并且在某些情况下,比如查询和更新操作频繁的情况下,它比线段树更快。

树状数组的核心思想是将数组的每个位置i表示为一个二进制数,然后用这个二进制数的最后一个1来确定它是哪个父节点的子节点。更具体地说,如果i的二进制表示中最后一个1位于第k位(从右往左数,起始位为0),那么i就属于它的父节点i + (1 << k)

下面是树状数组的C++实现:

#include <iostream>
#include <vector>

// 树状数组类
class BinaryIndexedTree {
private:
    int n;                  // 数组大小
    std::vector<int> tree;  // 树状数组

    //  lowbit函数,获取x的最后一个1的位置
    int lowbit(int x) {
        return x & -x;
    }

public:
    // 构造函数,初始化树状数组
    BinaryIndexedTree(int size) : n(size), tree(size + 1, 0) {}

    // 单点更新,将index位置的值增加delta
    void update(int index, int delta) {
        index++; // 树状数组的索引从1开始
        while (index <= n) {
            tree[index] += delta;
            index += lowbit(index);
        }
    }

    // 查询前缀和,即[1, index]的和
    int query(int index) {
        int sum = 0;
        index++; // 树状数组的索引从1开始
        while (index > 0) {
            sum += tree[index];
            index -= lowbit(index);
        }
        return sum;
    }

    // 查询区间和,即[left, right]的和
    int queryRange(int left, int right) {
        return query(right) - query(left - 1);
    }
};

int main() {
    // 测试代码
    BinaryIndexedTree bit(10); // 创建一个大小为10的树状数组
    bit.update(0, 1); // 将第1个元素更新为1
    bit.update(1, 2); // 将第2个元素更新为2
    bit.update(2, 3); // 将第3个元素更新为3
    std::cout << "Sum of elements from 1 to 3: " << bit.queryRange(1, 3) << std::endl; // 输出1到3的和
    return 0;
}

在这个实现中,我们使用一个数组tree来存储树状数组的节点值。update函数用于单点更新,它通过循环加上lowbit函数计算出的差值来更新所有受影响的节点。query函数用于查询前缀和,它通过循环减去lowbit函数计算出的差值来累加所有相关的节点值。queryRange函数用于查询区间和,它通过两个query调用来计算出区间和。

请注意,树状数组的索引是从1开始的,所以我们在所有操作中都对索引进行了加1的处理。此外,树状数组的大小通常是原始数组的大小加上1,以确保我们可以安全地使用索引来访问所有的节点。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值