樹(shù)狀數(shù)組是解決快速更新以及統(tǒng)計(jì)數(shù)組某段區(qū)間總和，設(shè)一個(gè)數(shù)組A[1-N],需要計(jì)算A[M-K]的總和，暴力解法需要O(K-M)，如果我們求出sum（1-K）和sum（1-M）,那么答案就是sum(1-M)-sum(1-K);

那么如何快速求出sum（1-N），可以考慮直接求，但如果我們?cè)偌右粋€(gè)條件，需要即時(shí)更新，那么使用暴力解法就會(huì)出現(xiàn)問(wèn)題，假設(shè)更新A[K]，需要對(duì)sum（1-K）到sum（1-N）進(jìn)行更新，這是一個(gè)非常費(fèi)時(shí)的過(guò)程。

這里我們使用一種二進(jìn)制的思想，我們將十進(jìn)制轉(zhuǎn)為二進(jìn)制進(jìn)行舉例思考，如上圖，

C[0001] = A[0001];

C[0010] = A[0010] + A[0001];

C[0011] = A[0011];

C[0100] = A[0100] + A[0011] + A[0010] + A[0001];

C[0101] = A[0101];

C[0110] = A[0110] + A[0101];

C[0111] = A[0111];

C[1000] = A[1000] + A[0111] + A[0110] + A[0101] + A[0100] + A[0011] + A[0010] + A[0001];

?把k定義為末尾連續(xù)0個(gè)數(shù)，可以理解為C[i] = A[i] + ... + A[i-2^k+1]。

更新的過(guò)程就是更新與A[i]相關(guān)的C[j]，舉例說(shuō)明：i = 0001，j = 0001, 0010, 0100, 1000，i = 0101, j = 0110, 1000；不難看出有不斷進(jìn)位的規(guī)律。

求sum得過(guò)程就是求取每一位1所對(duì)應(yīng)的數(shù)字的C[i]和，舉例說(shuō)明：

sum[1-1111] = C[1111] + C[1110] + C[1100] + C[1000]；

　　C[1111] = A[1111];

　　C[1110] = A[1110] + A[1101];

　　C[1100] = A[1100] + A[1011] + A[1010] + A[1001];

　　C[1000] = A[1000] + A[0111] + A[0110] + A[0101] + A[0100] + A[0011] + A[0010] + A[0001];

了解了樹(shù)狀數(shù)組所具有的二進(jìn)制規(guī)律，那么下來(lái)考慮如何用代碼實(shí)現(xiàn)。

int lowbit(x) {return x&-x; }void update(int k,int x) {for(int i=k;i<=n;i+=lowbit(i)) {C[i]+=x; } }

0001 -> 0010,?即 0001 + 0001 = 0010；

0010 -> 0100,?即 0010 + 0010 = 0100；

0101 -> 0110,?即 0101 + 0001 = 0110；

0110 -> 1000,?即 0110 + 0010 = 1000；

lowbit函數(shù)就是求取所加的最小進(jìn)位數(shù)，x&-x可以這樣理解:-x表示補(bǔ)碼，舉例可證原碼&補(bǔ)碼等于最小進(jìn)位數(shù)。

更新過(guò)程就是for循環(huán)從更新的k開(kāi)始，將所有C數(shù)組中相關(guān)元素更新。

int getsum(int x) {int ans=0;for(int i=x;i;i-=lowbit(i))//i要大于0ans+=C[i];return ans; }

1111 -> 1110,?即 1111 - 0001 = 1110；

1110 -> 1100,?即 1110 - 0010 = 1100；

1100 -> 1000, 即 1100 - 0100 = 1000；

1000 -> 0000,?即 1000 - 1000 = 0000；

同理可以看出求和是通過(guò)減去最小進(jìn)位數(shù)，得到下一個(gè)需要加的值。

轉(zhuǎn)載于:https://www.cnblogs.com/ACMessi/p/8469339.html

總結(jié)

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

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