阅读量:0
树状数组(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方法查询区间的和。