二叉索引树(树状数组)

二叉索引树 (Binary Indexed Tree,BIT)

      动态连续和查询问题,给定一个n个元素的数组A1,A2,..........An。支持以下两种操作

1.Add(x,d)操作:让Ax增加d。

2.Query(L,R):计算AL+AL+1+.......+AR。

     在学习二叉索引树之前,需要先介绍lowbit.

     对于正整数x,我们定义lowbit(x)为x的二进制表达式中最右边的1所对应的值。例如,38288的二进制是:1001010110010000,所以lowbit(x) = 16(二进制为10 000)。在程序实现中,lowbit(x) = x & (-x)。 因为计算机里的整数采用补码便是,因此-x实际上是x按位取反,末尾加1以后的结果。如下图


  

两者按位取“与”之后,前面部分全部为零,之后lowbit保持不变(都是第五位)。

下图是一棵典型的BIT,由十五个节点组成,编号为1 - 15.



灰色结点是BIT中的结点(白色长条的含义稍后叙述),每一层结点的lowbit相同,而且lowbit越大,越靠近根。图中的虚线是BIT中的边。编号为0的点是虚拟结点,他并不是树的一部分,但是他的存在可以让算法理解起来更容易一些。

对于结点i,如果它是左子结点,那么父结点的编号是i+lowbit(i)。如果它是右子结点,那么父结点的编号是i-lowbit(i)。搞清楚树的结构之后,构造一个辅助数组C,其中C[i]=A[i-lowbit(i)+1]+A[i-lowbit(i)+2]+....+A[i]。换句话说,C的每个元素都是A数组中的一段连续和。到底是那一段呢?在BIT中,每个灰色结点i都属于一个以它自身结尾的水平长条(对于lowbit=1的那些点,长条就是他本身),这个长条中数之和就是C[i]。其中

 C[i] = Ai-lowbit(x)+1 + Ai-lowbit(x)+2+...+Ai;

所以,有了C数组,就可以计算Si 了,其实只要节点i往左上“爬”就可以了,如下图



所以相应代码如下:

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

而如果要修改Ai,则应该顺着i点往右走,为什么?因为i点的改变会引起其父亲节点的改变。如下图



所对应的代码为:

void Add(int x,int d){
     while(x <= n){
       C[i] += d;
       x = x + lowbit(x);
     }
}
所以到最后我们就可以计算A1+A2+...+Ai,并且修改某Ai的值。


下面是一道练习题;

codevs P1080;

AC代码如下:

/*
作者:wbh20
题目:p1080 线段树练习
*/

#include<cstdio>

typedef long long LL;
LL C[100000+10],n,m;

inline int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int  res = 0;
	while(x > 0){
		res += C[x]; x -= lowbit(x);
	}
	return res;
}

void add(int x,int A){
	while(x <= n){
		C[x] += A; x += lowbit(x);
	}
}

int main(){
	scanf("%d",&n);
	for(int i = 1; i <= n; i++){
		int x;
		scanf("%d",&x);
		add(i,x);
	}
	scanf("%d",&m);
	while(m--){
		int cmd,a,b;
		scanf("%d%d%d",&cmd,&a,&b);
		if(cmd == 1)add(a,b);
		else printf("%d\n",sum(b) - sum(a-1));
	}
	return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值