1、概述

樹(shù)狀數(shù)組(binary indexed tree)，是一種設(shè)計(jì)新穎的數(shù)組結(jié)構(gòu)，它能夠高效地獲取數(shù)組中連續(xù)n個(gè)數(shù)的和。概括說(shuō)，樹(shù)狀數(shù)組通常用于解決以下問(wèn)題：數(shù)組{a}中的元素可能不斷地被修改，怎樣才能快速地獲取連續(xù)幾個(gè)數(shù)的和？

2、樹(shù)狀數(shù)組基本操作

傳統(tǒng)數(shù)組(共n個(gè)元素)的元素修改和連續(xù)元素求和的復(fù)雜度分別為O(1)和O(n)。樹(shù)狀數(shù)組通過(guò)將線性結(jié)構(gòu)轉(zhuǎn)換成偽樹(shù)狀結(jié)構(gòu)（線性結(jié)構(gòu)只能逐個(gè)掃描元素，而樹(shù)狀結(jié)構(gòu)可以實(shí)現(xiàn)跳躍式掃描），使得修改和求和復(fù)雜度均為O(lgn)，大大提高了整體效率。

給定序列（數(shù)列）A，我們?cè)O(shè)一個(gè)數(shù)組C滿足

C[i] = A[i–2^k+ 1] + … + A[i]

其中，k為i在二進(jìn)制下末尾0的個(gè)數(shù)，i從1開(kāi)始算！

則我們稱C為樹(shù)狀數(shù)組。

下面的問(wèn)題是，給定i，如何求2^k?

答案很簡(jiǎn)單：2^k=i&(i^(i-1)) ，也就是i&(-i)

下面進(jìn)行解釋：

以i=6為例（注意：a_x表示數(shù)字a是x進(jìn)制表示形式）：

(i)_10 = (0110)_2

(i-1)_10=(0101)_2

i xor (i-1) =(0011)_2

i and (i xor (i-1)) =(0010)_2

2^k = 2

C[6] = C[6-2+1]+…+A[6]=A[5]+A[6]

數(shù)組C的具體含義如下圖所示：

當(dāng)我們修改A[i]的值時(shí)，可以從C[i]往根節(jié)點(diǎn)一路上溯，調(diào)整這條路上的所有C[]即可，這個(gè)操作的復(fù)雜度在最壞情況下就是樹(shù)的高度即O(logn)。另外，對(duì)于求數(shù)列的前n項(xiàng)和，只需找到n以前的所有最大子樹(shù)，把其根節(jié)點(diǎn)的C加起來(lái)即可。不難發(fā)現(xiàn)，這些子樹(shù)的數(shù)目是n在二進(jìn)制時(shí)1的個(gè)數(shù)，或者說(shuō)是把n展開(kāi)成2的冪方和時(shí)的項(xiàng)數(shù),因此，求和操作的復(fù)雜度也是O(logn)。

樹(shù)狀數(shù)組能快速求任意區(qū)間的和：A[i] + A[i+1] + … + A[j]，設(shè)sum(k) = A[1]+A[2]+…+A[k]，則A[i] + A[i+1] + … + A[j] = sum(j)-sum(i-1)。

下面給出樹(shù)狀數(shù)組的C語(yǔ)言實(shí)現(xiàn)：

//求2^k int lowbit(int t) { return t & ( t ^ ( t - 1 ) ); } //求前n項(xiàng)和 int sum(int end) { int sum = 0; while(end > 0) { sum += in[end]; end -= lowbit(end); } return sum; } //增加某個(gè)元素的大小 void plus(int pos, int num) { while(pos <= n) { in[pos] += num; pos += lowbit(pos); }

3、擴(kuò)展——二維樹(shù)狀數(shù)組

一維樹(shù)狀數(shù)組很容易擴(kuò)展到二維，二維樹(shù)狀數(shù)組如下所示：

C[x][y] = sum(A[i][j])

其中，x-lowbit[x]+1 <= i<=x且y-lowbit[y]+1 <= j <=y

4、應(yīng)用

（1） 一維樹(shù)狀數(shù)組：

參見(jiàn)：http://hi.baidu.com/lilu03555/blog/item/4118f04429739580b3b7dc74.html

（2） 二維樹(shù)狀數(shù)組：

一個(gè)由數(shù)字構(gòu)成的大矩陣，能進(jìn)行兩種操作

1) 對(duì)矩陣?yán)锏哪硞€(gè)數(shù)加上一個(gè)整數(shù)（可正可負(fù)）

2) 查詢某個(gè)子矩陣?yán)锼袛?shù)字的和

要求對(duì)每次查詢，輸出結(jié)果

5、總結(jié)

樹(shù)狀數(shù)組最初是在設(shè)計(jì)壓縮算法時(shí)發(fā)現(xiàn)的（見(jiàn)參考資料1），現(xiàn)在也會(huì)經(jīng)常用語(yǔ)維護(hù)子序列和。它與線段樹(shù)（具體見(jiàn)：數(shù)據(jù)結(jié)構(gòu)之線段樹(shù)）比較在思想上類似，比線段樹(shù)節(jié)省空間且編程復(fù)雜度低，但使用范圍比線段樹(shù)小（如查詢每個(gè)區(qū)間最小值問(wèn)題）。

6、參考資料

（1） Binary Indexed Trees：

http://www.topcoder.com/tc?module=Static&d1=tutorials&d2=binaryIndexedTrees

（2） 吳豪文章《樹(shù)狀數(shù)組》：

http://www.java3z.com/cwbwebhome/article/article19/zip/treearray.zip

（3） 郭煒文章《線段樹(shù)和樹(shù)狀數(shù)組》：

http://poj.org/summerschool/1_interval_tree.pdf

總結(jié)

以上是生活随笔為你收集整理的树桩数组的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。