#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
*/
树状数组
最新推荐文章于 2023-06-21 22:43:57 发布