洛谷P3368 树状数组2

最新推荐文章于 2022-03-08 17:29:28 发布

海将河推走

最新推荐文章于 2022-03-08 17:29:28 发布

阅读量352

点赞数

分类专栏：数据结构和算法文章标签： c++ 数据结构算法

本文链接：https://blog.csdn.net/weixin_43669941/article/details/108912989

版权

这篇博客详细介绍了洛谷P3368题目的解决方案，重点讲解了树状数组的概念、作用以及C++的实现。同时，解释了差分数组的原理，指出利用树状数组维护差分数组能高效地处理区间加法问题，并提供了相应的C++代码示例。

摘要由CSDN通过智能技术生成

洛谷P3368 树状数组2

首先简单介绍一下树状数组，以及为什么要使用树状数组。

假设有一个数组，我们需要对该数组执行如下操作（以下全部假设数组下标从1开始）：

求前i项的和。
将第i项元素加上某个给定的值。

如果使用普通的数组，执行第一个操作的时间为 $O (n)$ ，执行第二个操作的时间为 $O (1)$ 。

如果用数组保存的内容为前缀和，执行第一个操作的时间为 $O (1)$ ，执行第二个操作的时间为 $O (n)$ 。

而使用树状数组，这两个操作的时间都为 $O(\log_2n)$ 。

关于树状数组的原理，请观看上面链接中的视频或寻找其他资料。

下面是C++的简单实现代码：

int n;
int arr[500001];

inline int last_bit(int num) {
   
    return num & (-num);
}

// return the sum of [1, index]
int sum(int index) {
   
    int res = 0;
    while (index) {
   
        res += arr[index];
        index -= last_bit(index);
    }

    return res;
}

// plus value to the element of index of arr
void add(int index, int value) {
   
    while (index <= n) {
   
        arr[index] += value;
        index += last_bit(index);
    }
}