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#中,可以使用递归方法来获取所有控件。下面是一个示例代码:
    public List GetAllControls(Control container)
    { List controlList = new List(); ...

  • c# winform如何开发程序

    要开发一个C# WinForm程序,可以按照以下步骤进行: 打开Visual Studio,选择创建一个新的项目。
    在项目模板中选择Windows Forms App (.NET Framework)。

  • c#和java哪个简单

    这个问题没有一个明确的答案,因为一个人认为简单的语言可能对另一个人来说很困难。C#和Java都是流行的编程语言,它们在某些方面相似,但在其他方面又有所不同。...

  • c#怎么比较两个日期

    在C#中,可以使用DateTime.Compare方法来比较两个日期。这个方法会返回一个整数值,表示两个日期之间的比较结果。
    下面是一个示例代码,演示了如何比较两个...

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

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

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

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

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

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

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

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