树状数组算法

树状数组，是一种快速解决单点更新区间查询(或区间更新单点查询)的数据结构.
设$a_1, a_2, a_3, ..., a_n$为长度为n($\le10^5)$的数列. 现在有以下两种类型的操作m($\le10^4$)个:

1. ADD A B (修改数列中A位置上的元素值增加B)
2. QUERY A B (询问区间[A, B]的元素值和)

直接求解： O(1)的更新，m次O(n)的查询复杂度. O(mn) 不能接受.

引入辅助数组C. 其中:
$C_1 = a_1$
$C_2 = a_1 + a_2$
$C_3 = a_3$
$C_4 = a_1 + a_2 + a_3 + a_4$
$C_5 = a_5$
$C_6 = a_5 + a_6$
$C_7 = a_7$
$C_8 = a_1 + a_2 + a_3 + a_4 + a_5 + a_6 + a_7 + a_8$
$...$
$C_{16} = a_1 + a_2 + a_3 + a_4 + a_5 + a_6 + a_7 + a_8+ a_9 + a_{10} + a_{11} + a_{12} + a_{13} + a_{14} + a_{15} + a_{16}$

$对于任意的n, C_n = a_{n-2^k+1} + a_{n-2^k+2} + ... + a_n = \sum_{i=n-2^k+1}^n a_i, 其中k为n二进制表示中末尾0的个数$

对于数组C可以构造一棵树, 如下图所示(图片摘自百度百科):

上图中由C数组构造的一棵树满足: $对于节点C_i和C_j, 如果j+2^k == i, 则C_i是C_j的父节点(其中k为j的末尾0个数)$
设$lowbit(x) = 2^k (k为x二进制表示末尾0的个数).$

1. 求区间[A, B]元素和
   定义$sum[i] = a_1 + a_2 + ... + a_i$, 问题转化成求sum[B] - sum[A - 1]. 推倒sum[i]的求法：
   

   $$\begin{align} sum[i] &= a_1 + a_2 + ... + a_i\\ &=a_1 + a_2 + ... + a_{i-2^k} + a_{i-2^k+1} + ... + a_i\\ &=a_1 + a_2 + ... + a_{i-2^k} + C_i, (2^k == lowbit(i))\\ &=a_1 + a_2 + ... + a_{i - lowbit(i)} + C_i\\ \end{align}$$
   因而计算$sum[i]$的过程如下代码所示:

   int lowbit(int x){
       return x & (-x);
   }
   
   int sum(int n){
       int ans = 0;
       while(n > 0){
           ans += c[n];
           n -= lowbit(n);
       }
       return ans;
   }

   计算sum[i]的过程实则为去除i末尾1的过程, 因而算法复杂度为O(logn)
   
   

2. 单点更新位置i上的元素增加d.
   当更新$a_i$时，从树的性质能看出，只需要更新$C_i$及其所有的父节点$C_{i+lowbit(i)} . . .等$

   void add(int i, int d){
       while(i < MAX){
           c[i] += d;
           i += lowbit(i);
       }
   }
   

   由于树的高度最多为log(n). 因而更新维护C数组的复杂度也是O(logn).

树状数组解决的问题举例:

1. 单点更新，区间查询.
   最原始的算法套用. 查询[A, B]区间和, ans = sum[B] - sum[A - 1]

2. 区间更新，单点查询:
   【例】
   a) ADD A B d (将区间[A, B]中的元素增加d)
   b) QUERY x (查询x的值)
   这里只需把a)操作转化为add(A, d), add(B + 1, -d). 即可.

3. 二维树状数组:
   【例】 二维矩阵M*N(M, N <=10^3), 有以下两种操作
   a) ADD x y d(将x,y 位置的元素增加d)
   b) QUERY x1 y1 x2 y2( 查询矩形区域(x1,y1)(左上角), (x2, y2)(右下角)的和)
   计算ans = sum(x2, y2) - sum(x2, y1 - 1) - sum(x1 - 1, y2) + sum(x1 - 1, y1 - 1)
   二维树状数组:

   void add(int x, int y, int d){
       for(int i = x; i <= n; i += lowbit(i)){
           for(int j = y; j <= n; j += lowbit(j)){
               c[i][j] += d;
           }
       }
   }
   
   int sum(int x, int y){
       int ans = 0;
       for(int i = x; i > 0; i -= lowbit(i)){
           for(int j = y; j > 0; j -= lowbit(j)){
               ans += c[i][j];
           }
       }
       return ans;
   }