树状数组原理及操作

lowbit 操作原理如下所示，lowbit即取非负整数n在二进制表示下最低位1及其后面的0构成的数值

 

 

树状数组是动态维护前缀和的工具，可以有效进行单点查询（ask)，以及修改和区间查询（add)

摘自宫水三叶总结的树状数组模板

tree[i]保存以x为根的子树中节点值的和，i<=n

树状数组如下图所示

 

 

将每个节点转换成二进制后，我们会发现每一层的末尾的零的个数是相同的，而且末尾零的个数与其覆盖的长度有关，并且t[i]节点的父节点为t[i+lowbit(i)]（在单点查询修改用到），单点查询修改add()方法如下所示

 

ask 查询数组序号x之前的所有和，查询这个点的前缀和，需要从这个点向左上找到上一个节点，

如何找左上节点  i=i-lowbit(i)
把当前节点的值和所有左上节点的值进行相加，即该点的前缀和
如果需要求区间和，即分别求出前缀和并相减,原理图如下所示

总体模板 

//ask 查询数组序号x之前的所有和，查询这个点的前缀和，需要从这个点向左上找到上一个节点，
//把当前节点的值和所有左上节点的值进行相加，即该点的前缀和
//如果需要求区间和，即分别求出前缀和并相减

int ask(int x){
int ans=0;
for(int i=x;i>0;i=i-lowbit(i)){
ans+=tree[i];
}
return ans;
}

整棵树的深度为log2n+1

// 上来先把三个方法写出来
{
    int[] tree;
//lowbit即取非负整数n在二进制表示下最低位1及其后面的0构成的数值
    int lowbit(int x) {
//相反数为x的反码加+1，再进行&操作，得到期望值
        return x & -x;
    }
    // 查询前缀和的方法
    int query(int x) {
        int ans = 0;
        for (int i = x; i > 0; i -= lowbit(i)) ans += tree[i];
        return ans;
    }
    // 在树状数组 x 位置中增加值 u
    void add(int x, int u) {
         //向上找父节点，只需要将下标+=lowbit(这个点的下标)，子节点的值要加u
         //为了有效维护，其所有的父节点，及“爷爷”节点等都要加U
        for (int i = x; i <= n; i += lowbit(i)) tree[i] += u;
    }
}

// 初始化「树状数组」，要默认数组是从 1 开始
{
    for (int i = 0; i < n; i++) add(i + 1, nums[i]);
}

// 使用「树状数组」：
{   
    void update(int i, int val) {
        // 原有的值是 nums[i]，要使得修改为 val，需要增加 val - nums[i]
        add(i + 1, val - nums[i]); 
        nums[i] = val;
    }
    //查询数组区间和
    int sumRange(int l, int r) {
        return query(r + 1) - query(l);
    }
}