树状数组实现

树状数组能够完成如下操作:

给一个序列a0-an

计算前i项和

对某个值加x

时间o(logn)

 

注意:有人觉得前缀和就行了,但是你还要维护啊,改变某个值,一个一个改变前缀和就是o(n)了。

线段树树状数组的题就是这样,维护一个树,比较容易看出来。

 

 

线段树:

https://blog.csdn.net/hebtu666/article/details/82691008

如果使用线段树,只需要对网址中的实现稍微修改即可。以前维护最小值,现在维护和而已。

注意:要求只是求出前i项,而并未给定一个区间,那我们就能想出更快速、方便的方法。

对于任意一个节点,作为右孩子,如果求和时被用到,那它的左兄弟一定也会被用到,那我们就没必要再用右孩子,因为用他们的父就可以了。

这样一来,我们就可以把所有有孩子全部去掉

把剩下的节点编号。

如图,可以发现一些规律:1,3,5,7,9等奇数,区间长度都为1

6,10,14等长度为2

........................

如果我们吧编号换成二进制,就能发现,二进制以1结尾的数字区间长度为1,最后有一个零的区间为2,两个零的区间为4.

我们利用二进制就能很容易地把编号和区间对应起来。

 

计算前i项和。

需要把当前编号i的数值加进来,把i最右边的1减掉,直到i变为0.

二进制最后一个1可以通过i&-i得到。

 

更新:

不断把当前位置i加x,把i的二进制最低非零位对应的幂加到i上。

下面是代码:

思想想出来挺麻烦,代码实现很简单,我都不知道要注释点啥

向发明这些东西的大佬们致敬

int bit[MAX_N+1]
int n;

int sum(int i)
{
    int gg=0;
    while(i>0)
    {
        gg+=bit[i];
        i-=i&-i;
    }
    return gg;
}

void add(int i,int x)
{
    while(i<=n)
    {
        bit[i]+=x;
        i+=i&-i;
    }
}

 

  • 172
    点赞
  • 201
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论
树状数组是一种用于快速维护数组前缀和的数据结构,其时间复杂度为 $O(logn)$。树状数组实现基于二进制的思想,借助树状数组可以高效地进行单点更新和区间查询等操作。以下是树状数组实现原理: 1. 数组的每个元素代表原数组中某个位置的前缀和,如下所示: $$C_i = A_1 + A_2 + \cdots + A_i$$ 其中 $C_i$ 表示原数组 $A$ 的前 $i$ 项之和。 2. 将 $C$ 数组转换成树状数组 $T$,树状数组的每个节点表示其父节点到该节点的元素和。如下所示: $$T_i = C_{i-lowbit(i)+1} + C_{i-lowbit(i)+2} + \cdots + C_i$$ 其中 $lowbit(i)$ 表示 $i$ 的二进制表示中最低位的 $1$ 所代表的值。 3. 树状数组的单点更新操作只需要将 $i$ 位置的值加上 $k$,然后依次更新其祖先节点即可: ```cpp void update(int i, int k, int n, vector<int>& tree) { while (i <= n) { tree[i] += k; i += lowbit(i); } } ``` 其中 $n$ 表示原数组 $A$ 的大小,$tree$ 表示树状数组。 4. 树状数组的区间查询操作只需要通过两次前缀和的差值计算出区间和即可: ```cpp int query(int i, vector<int>& tree) { int sum = 0; while (i > 0) { sum += tree[i]; i -= lowbit(i); } return sum; } int query(int i, int j, vector<int>& tree) { return query(j, tree) - query(i - 1, tree); } ``` 其中第一个查询函数用于计算 $A_1 + A_2 + \cdots + A_i$,第二个查询函数用于计算 $A_{i+1} + A_{i+2} + \cdots + A_j$,然后两者相减即可得到区间 $[i,j]$ 的和。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

兔老大RabbitMQ

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值