温馨提示×

利用C++树状数组处理大规模数据

c++
小樊
83
2024-08-19 20:54:35
栏目: 编程语言

树状数组(Fenwick Tree)是一种用于高效处理动态区间和的数据结构,可以用来解决一些统计问题,比如求区间和、区间最大值、区间最小值等。在处理大规模数据时,树状数组可以帮助我们高效地进行数据统计和查询。

下面是一个使用C++实现树状数组的示例代码:

#include <iostream>
#include <vector>

using namespace std;

// 树状数组的实现
class FenwickTree {
public:
    FenwickTree(int n) {
        tree.resize(n + 1, 0);
    }

    void update(int index, int delta) {
        while (index < tree.size()) {
            tree[index] += delta;
            index += index & -index;
        }
    }

    int query(int index) {
        int sum = 0;
        while (index > 0) {
            sum += tree[index];
            index -= index & -index;
        }
        return sum;
    }

private:
    vector<int> tree;
};

int main() {
    int n = 10; // 数据规模
    FenwickTree tree(n);

    // 更新元素
    tree.update(1, 2);
    tree.update(2, 3);
    tree.update(3, 5);

    // 查询区间和
    cout << tree.query(3) << endl; // 输出:10
    cout << tree.query(5) - tree.query(2) << endl; // 输出:8

    return 0;
}

在上面的示例代码中,我们实现了一个简单的树状数组类FenwickTree,并使用其来处理一些数据更新和求和操作。通过树状数组,我们可以高效地处理大规模数据,并快速进行数据统计和查询。

0