码迷,mamicode.com
首页 > 编程语言 > 详细

树状数组

时间:2020-01-23 09:37:07      阅读:76      评论:0      收藏:0      [点我收藏+]

标签:amp   begin   mic   return   define   inf   span   计算   截取   

树状数组是一种有效更新元素并计算前缀和的数据结构。
给定一个数组,需要进行上述操作。朴素算法是线性时间复杂度,而树状数组允许在O(logn)的时间内执行这两个操作。

假设原数组为a,与之等价的树状数组为c,有如下关系:
$\begin{cases} \ c[x]=\sum_{i=x-lowbit(x)+1}^{x}a[i] \\ \ lowbit(x)=x\&(-x) \end{cases}$
lowbit(x)截取x的二进制形式中最低位的1及其后面的0,因此x-lowbit(x)可以抹去x的二进制形式中最低位的1。
比如(以下数值均为二进制),lowbit(10110)=10,10110-lowbit(10110)=10100。
等价关系的图像形式如图所示:

技术图片

其中每个结点的值都等于其子孙结点值的和。
由图可以看出更新操作只需不断向上修改父结点,由关系式得出计算前缀和操作只需以lowbit为跨度求c数组的和。既然操作是抬升或消除二进制中的尾1,那么总操作次数必定小于等于二进制位数,这是对数级别的。

#define lowbit(x) x&-x
const int S;
int c[S] = {0};
void add(int x, int d){
    while(x <= S) c[x] += d,x += lowbit(x);
}
int sum(int x){
    int ret = 0;
    while(x > 0) ret += c[x], x -= lowbit(x);
    return ret;
}

树状数组

标签:amp   begin   mic   return   define   inf   span   计算   截取   

原文地址:https://www.cnblogs.com/nioh/p/12229987.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!