最近再看树状数组,开始有些不太理解,今天终于弄懂了。
对于 lowbit
计算的如此巧妙真是太神奇了,一行代码即可获取一个数最后一位1的位置
int lowbit(int x) {
return x & -x;
}
但是树状数组的思想前几天一直困扰我,在花了这个图之后慢慢明白了。记录一下:
其实树桩数组就是将几个书的和存起来,这样计算n个数和的时候复杂度就可以从N到log(N),通过二进制思想,确定了那几个元素的和存到那个元素上。分的时候,一直找最低位1的值,如上图所示
简单的添加和查找操作如下。
#define LOCAL
#include <stdio.h>
#include <iostream>
#include <string.h>
using namespace std;
const int maxn = 100000 + 5;
int d[maxn], n;
int lowbit(int x) {
return x & -x;
}
int query(int x) {
int res = 0;
while (x) {
res += d[x];
x -= lowbit(x);
}
return res;
}
// 对数状态数组的更新
int add(int x, int v) { // 对 x 加 v
while (x <= n) {
d[x] += v;
x += lowbit(x);
}
}
int main() {
#ifdef LOCAL
freopen("in.txt", "r", stdin);
#endif
return 0;
}