借用一下y总的图片
树状数组中每一个节点都是存储了一段区间的值
一个节点在第 k 层是这个节点的下标有几个2^k次方
例如:
4 = 2 ^ 2 所以4是第二层
3 = 2 ^ 0 * 3 所以是0层
12 = 2 ^ 2 * 3 所以是第二层
这里就运用了lowbit算法传入一个正整数会返回这个数二进制的最后一个1也就是最大有2 的 几次方
下面是一道树状数组的题目acwing1215代码
c++
#include<iostream>
#include<cstring>
using namespace std;
const int N = 100010, M = 1000010;
int n;
int a[N] = {0}, sum[N] = {0}, tr[M] = {0};
int lowbit(int x){
return x & -x;
}
void add(int x, int v){//将可以将x下标作为子节点的下标全部加上v
//例如1是2的子节点
for(int i = x; i <= M; i += lowbit(i)) tr[i] += v;
}
int query(int x){//查询x为下标的前缀和
int res = 0;
for(int i = x; i >= 1; i -= lowbit(i)) res += tr[i];
return res;
}
int main(){
cin >> n;
for(int i = 1; i <= n; i++) cin >> a[i], a[i]++;
for(int i = 1; i <= n; i++){
sum[i] += query(M - 1) - query(a[i]); //右边比这个数大的有几个 加上了一个前缀和
add(a[i], 1);
}
memset(tr, 0, sizeof tr);
for(int i = n; i >= 1; i--){
sum[i] += query(a[i] - 1);//左边比这个数小的有几个 加上了一个前缀和
add(a[i], 1);
}
long long res = 0;
for(int i = 1; i <= n; i++){
res += (long long)sum[i] * (sum[i] + 1) / 2;
}
cout << res;
}