1. 程式人生 > >【專題】樹狀陣列(完整版)

【專題】樹狀陣列(完整版)

傳統陣列(共n個元素)的元素修改和連續元素求和的複雜度分別為O(1)和O(n)。樹狀陣列通過將線性結構轉換成偽樹狀結構(線性結構只能逐個掃描元素,而樹狀結構可以實現跳躍式掃描),使得修改和求和複雜度均為O(lgn),大大提高了整體效率。

給定序列(數列)A,我們設一個數組C滿足

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

其中,k為i在二進位制下末尾0的個數,i從1開始算!

則我們稱C為樹狀陣列。

下面的問題是,給定i,如何求2^k?

答案很簡單:2^k=i&(i^(i-1)) ,也就是i&(-i)    為什麼呢?? 請看下面:

整數運算 x&(-x),當x為0時結果為0;x為奇數時,結果為1;x為偶數時,結果為x中2的最大次方的因子。
       因為:x &(-x) 就是整數x與其相反數(負號取反)的按位與:1&1=1,0&1 =0, 0&0 =1。具體分析如下:
       □ 當x為0時,x&(-x) 即 0 & 0,結果為0;
       □ 當x不為0時,x和-x必有一個為正。不失一般性,設x為正。
       ●當x為奇數時,最後一個位元為1,取反加1沒有進位,故x和-x除最後一位外前面的位正好相反,按位與結果為0。最後一位都為1,故結果為       1。
       ●當x為偶數,且為2的m次方(m>0)時,x的二進位制表示中只有一位是1(從右往左的第m+1位),其右邊有m位0,左邊也都是0(個數由表示   x的字        節數決定),故x取反加1後,從右到左第有m個0,第m+1位及其左邊全是1。這樣,x& (-x) 得到的就是x。 
       ●當x為偶數,卻不為2的m次方的形式時,可以寫作x= y * (2^k)。其中,y的最低位為1。實際上就是把x用一個奇數左移k位來表示。這時,x的   二進位制         表示最右邊有k個0,從右往左第k+1位為1。當對x取反時,最右邊的k位0變成1,第k+1位變為0;再加1,最右邊的k位就又變成了0,第   k+1位因為進         位的關係變成了1。左邊的位因為沒有進位,正好和x原來對應的位上的值相反。二者按位與,得到:第k+1位上為1,左邊右邊都為   0。結果為2^k,即         x中包含的2的最大次方的因子。
        總結一下:x&(-x),當x為0時結果為0;x為奇數時,結果為1;x為偶數時,結果為x中2的最大次方的因子。 比如x=32,其中2的最大次方因子  為                 2^5,故x&(-x)結果為32;當x=28,其中2的最大次方因子為4,故x & (-x)結果為4。當x=24,其中2的最大次方因子為8,故 x&(-x)結果為  8。


下面進行解釋:

以i=6為例(注意:a_x表示數字a是x進製表示形式):

(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]

陣列C的具體含義如下圖所示:

當我們修改A[i]的值時,可以從C[i]往根節點一路上溯,調整這條路上的所有C[]即可,這個操作的複雜度在最壞情況下就是樹的高度即O(logn)。另外,對於求數列的前n項和,只需找到n以前的所有最大子樹,把其根節點的C加起來即可。不難發現,這些子樹的數目是n在二進位制時1的個數,或者說是把n展開成2的冪方和時的項數,因此,求和操作的複雜度也是O(logn)。

樹狀陣列能快速求任意區間的和:A[i] + A[i+1] + … + A[j],設sum(k) = A[1]+A[2]+…+A[k],則A[i] + A[i+1] + … + A[j] = sum(j)-sum(i-1)。