树状数组学习（例题及题解）

最新推荐文章于 2025-07-21 20:52:59 发布

qq_34026804

最新推荐文章于 2025-07-21 20:52:59 发布

阅读量481

点赞数

CC 4.0 BY-SA版权

分类专栏：模板学习

本文链接：https://blog.csdn.net/qq_34026804/article/details/115187250

本文介绍了树状数组的概念及其在动态维护前缀和中的应用。通过代码模板展示了如何实现添加元素和查询前缀和的操作，并提供了两个实际问题的解题思路：一是楼兰图腾问题，利用树状数组计算特定位置的乘积；二是整数问题，结合差分数组进行动态更新。这些例子展示了树状数组在解决数学和数组操作问题上的高效性。

在这里插入图片描述
树状数组：可以动态维护前缀和，查询时间复杂度O（log₂n）,修改时间复杂度O（log₂n）。
树状数组记录前缀和的方式为这个数的二进制表示中从长度是从右往左的第一个0的大小,即长度为x-lowbit(x)+1。
代码模板

int tr[N];
int lowbit(int x)   //求第一个0的位置
{
    return x & -x;
}

void add(int x, int c) 在位置为x的地方加上c
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}

int sum(int x) 求x的前缀和
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

树状数组题解
1.楼兰图腾

大致思路：对于点i，求出左边和右边比i小的个数相乘，即为在i点∧的个数，同理，求出左边和右边比i大的个数相乘，即为在i点V的个数。
故可以用两个数组记录
Low[i]表示左边比第i个位置小的数的个数
Great[i]表示左边比第i个位置大的数的个数
tr[i]求和表示用于记录小于等于i个数的总和