117.info
人生若只如初见

如何在C++中实现树状数组

树状数组(Binary Indexed Tree,BIT)是一种用于动态维护序列前缀和的数据结构,可以在O(logn)的时间复杂度内完成区间和的更新和查询操作。以下是在C++中实现树状数组的示例代码:

#include 
#include 

using namespace std;

class FenwickTree {
private:
    vector tree;

public:
    FenwickTree(int n) {
        tree.resize(n + 1);
    }

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

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

int main() {
    vector nums = {1, 3, 5, 7, 9, 11};
    FenwickTree tree(nums.size());

    for (int i = 0; i < nums.size(); i++) {
        tree.update(i + 1, nums[i]);
    }

    cout << "Prefix sum of first 3 elements: " << tree.query(3) << endl;
    cout << "Prefix sum of elements in range [2, 5]: " << tree.query(5) - tree.query(1) << endl;

    return 0;
}

在这个示例中,我们首先定义了一个FenwickTree类,用于实现树状数组的功能。在主函数中,我们创建了一个FenwickTree对象,并使用update方法更新了树状数组的值。最后,我们使用query方法查询了前缀和的结果。

这是一个简单的示例,实际应用中可以根据具体需求对树状数组进行进一步的封装和优化。

未经允许不得转载 » 本文链接:https://www.117.info/ask/fea5aAzsAAwNTAQ.html

推荐文章

  • c++数组的定义方式有哪些

    C++数组的定义方式有以下几种: 声明并初始化数组:可以在声明数组的同时初始化数组元素。例如: int arr1[] = {1, 2, 3, 4, 5}; // 自动根据初始化列表确定数组...

  • C++怎么实现共享内存

    C++中可以使用System V共享内存和POSIX共享内存来实现共享内存。 使用System V共享内存: 首先,需要包含和头文件。 调用shmget()函数创建或获取共享内存标识符。...

  • C#中Directory.GetFiles()函数怎么使用

    Directory.GetFiles()函数用于获取指定目录中的所有文件的路径,可以通过传入不同的参数来指定要搜索的目录、搜索的文件类型以及搜索的方式等。
    以下是Dire...

  • c++如何判断文件是否存在

    在C++中,可以使用std::ifstream类来判断文件是否存在。以下是一个示例代码:
    #include #include bool fileExists(const std::string& filename) {
    st...

  • C++树状数组的基本原理是什么

    C++树状数组(Binary Indexed Tree,BIT)是一种用于高效实现动态维护动态序列的数据结构,能够支持动态的单点更新和区间查询。树状数组的基本原理是通过对原始序...

  • 时间戳在C++多线程编程中的应用

    时间戳在C++多线程编程中有多种应用,其中一些常见的用途包括: 记录事件发生的时间:在多线程编程中,我们经常需要记录事件发生的时间,以便分析程序的执行顺序...

  • C++时间戳的精度损失问题分析

    在C++中,通常使用std::chrono::system_clock来获取当前时间戳,其精度往往取决于操作系统和硬件平台的支持。一般来说,std::chrono::system_clock提供的精度为纳...

  • 利用C++时间戳实现日志记录功能

    在C++中,可以使用time库来获取当前时间戳,并将时间戳转换成可读的日期时间格式,然后将日志记录到文件中。以下是一个简单的示例代码:
    #include #include...