树状数组的三个基本函数

上次听强哥讲完之后没怎么复习过,趁着放假时间多赶紧看看


核心代码有三:

一, 核心lowbit函数,得到一个数的二进制最低位的1,代码很简单,但是是最核心的代码

int lowbit(int x)
{
    return x & (-x);
}

例如10的二进制是1010,返回值应该为2即二进制的10


二,求和函数getsum

int getsum(int x)
{
    int sum = 0;
    while(x > 0)
    {
        sum += c[x];
        x -= lowbit(x);
    }
    return sum;
}

例如求前12个元素的和,模拟一下,应该是

sum = 0, x = 12

sum += c[12], x -= 4, x = 8

sum += c[8], x -= 8, x = 0 

然后x == 0 时结束循环


三, 更新元素的函数update

void update(int x, int num)
{
    while(x <= n)
    {
        c[x] += num;
        x += lowbit(x);
    }
}

在数组某个元素进行更新的时候效率很高,只要更新包含变动元素的c[]数组即可,先更新该元素,然后不断向上更新


树状数组与二进制有着密不可分的联系,理解起来并不困难,代码实现也比较容易,效率也是相当的高

试着去找题做一做




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值