树状数组

#include<iostream>
using namespace std;
/*
		树状数组 / BIT / Binary Index Tree
		求前缀和 / 单点增加
*/
//原数列
int a[1000];
//树状数组,BIT[i]表示区间a[i - lowbit(i) + 1]至a[i]的和(闭)
int BIT[1000];
int n;//length of a
//lowbit运算,取当前最低位的1及其低位0组成的数,lowbit(1100B) = 100B
int lowbit(int x) {
	return x & -x;
}
int _lowbit(int x) {
	//在补码中,[-x]补码 = [x]补码求反+1,所以以下操作等价 x & -x
	return x & (~x + 1);
}
//求前缀和a[1]~a[x]
int get(int x) {
	int ans = 0;
	while (x) {
		ans += BIT[x - lowbit(x)];
		x -= lowbit(x);
	}
	return ans;
}
//单点增加,为a[i]加上x时要正确维护前缀和
void add(int i, int x) {
	while (i <= n) {
		BIT[i] += x;
		i += lowbit(i);
	}
}
int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		add(i, a[i]);
	}
	for (int i = 1; i <= n; i++) {
		cout << BIT[i] << endl;
	}
	return 0;
}
/*
		对于序列1,2,3,4,5,6,7,8,9,10
		a[1] ~ a[10] : 1,2,3,4,5,6,7,8,9,10
		BIT[10]: [1000B + 1 ~ 1010B] 即 a[9] ~ a[10]sum = 19
		BIT[9] : [1000B + 1 ~ 1001B] 即 a[9] ~ a[9] sum = 9
		BIT[8] : [0000B + 1 ~ 1000B] 即 a[1] ~ a[8] sum = 36
		BIT[7] : [0110B + 1 ~ 0111B] 即 a[7] ~ a[7] sum = 7
		BIT[6] : [0100B + 1 ~ 0110B] 即 a[5] ~ a[6] sum = 11
		BIT[5] : [0100B + 1 ~ 0101B] 即 a[5] ~ a[5] sum = 5
		BIT[4] : [0000B + 1 ~ 0100B] 即 a[1] ~ a[4] sum = 10
		BIT[3] : [0010B + 1 ~ 0011B] 即 a[3] ~ a[3] sum = 3
		BIT[2] : [0000B + 1 ~ 0010B] 即 a[1] ~ a[2] sum = 3
		BIT[1] : [0000B + 1 ~ 0001B] 即 a[1] ~ a[1] sum = 1
*/

/*
	input
10
1 2 3 4 5 6 7 8 9 10
	---
	output
1
3
3
10
5
11
7
36
9
19
*/
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值