利用C++树状数组解决区间查询问题

   2024-10-01 9710
核心提示:树状数组(Binary Indexed Tree)是一种用于高效处理区间查询的数据结构。在C++中,可以通过以下代码实现树状数组:#include ios

树状数组(Binary Indexed Tree)是一种用于高效处理区间查询的数据结构。在C++中,可以通过以下代码实现树状数组:

#include <iostream>#include <vector>using namespace std;class FenwickTree {private:    vector<int> tree;public:    FenwickTree(int n) {        tree.resize(n + 1);    }    void update(int index, int val) {        while (index < tree.size()) {            tree[index] += val;            index += (index & -index);        }    }    int query(int index) {        int sum = 0;        while (index > 0) {            sum += tree[index];            index -= (index & -index);        }        return sum;    }    int rangeQuery(int left, int right) {        return query(right) - query(left - 1);    }};int main() {    vector<int> nums = {1, 3, 5, 7, 9, 11, 13};    FenwickTree fenwickTree(nums.size());    for (int i = 0; i < nums.size(); i++) {        fenwickTree.update(i + 1, nums[i]);    }    cout << "Sum of elements in range [2, 5]: " << fenwickTree.rangeQuery(2, 5) << endl;    return 0;}

在这段代码中,我们首先定义了一个FenwickTree类,其中包含了update和query两个方法用于更新和查询操作。在主函数中,我们创建了一个FenwickTree对象,将给定数组的元素依次插入到树状数组中。最后,我们可以使用rangeQuery方法查询区间的和。

 
举报打赏
 
更多>同类物流大全
推荐图文
推荐物流大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号