树状数组求区间和

Reference: 树状数组解析

1.统计区间 sum

2.修改某个数组值

#define lowbit(i) ( -i  &  i) // 二进制数字最低位'1'和其后'0'组成的数字
template<typename T>
class BinaryIndexedTree {
public:
	BinaryIndexedTree(const vector<T>& arr) :LEN(arr.size()){
		BuildBIT(arr);
	}
	~BinaryIndexedTree() {
	}
	/* 
	 * 构建树状数组,index 从1开始, 输入数组index 从 0 开始,利用update 操作完成
	 */
	void BuildBIT(const vector<T> &arr) {
		bit.resize(LEN+ 1, 0);
		for (auto i = 1; i <= LEN; ++i)
			update(i, arr[i - 1]);
	}
	/*
	 * 直观上是,应该更新 arr[i] 
	 * 实际上,更新树状数组, val为增量或减量,而不是更新后的值 
	 */
	void update(int i, int val) {
		while (i <= LEN) {
			bit[i] += val;
			i += lowbit(i); // 类似父结点,从而更新父亲结点
		}
	}
	/*
	*  累加arr[1 .. index] 的值 并且返回
	*/
	T sum( int index) {
		T ret = 0;
		while (index > 0) {
			ret += bit[index];
			index -= lowbit(index); //类似兄弟结点
		}
		return ret;
	}
	/*
	* 累加arr[index_i + 1 .. index_j] 的值 并且返回
	*/
	T sum( int index_i,  int  index_j) {
		return sum(index_j) - sum(index_i);
	}
	/*
	 * 返回 arr[index]
	 * 三种方法: 
	 *					1. 增加一个arr数组
	 *					2. 利用 sum(index - 1, index);
	 *					3. 当前方法
	 */
	T getElem(int index) {
		T ret = bit[index];
		unsigned sibling = index - lowbit(index);
		index--;
		while (index != sibling) {
			ret -= bit[index];
			index -= lowbit(index); //类似兄弟结点

		}
		return ret;
	}
private:
	vector<T> bit;
	const unsigned LEN;
};

树状数组结构: 

1.黑色代表原有数组,红色代表树状数组

2. 有同一父结点的结点为兄弟结点



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值