树状数组总结【全】

【前言】

 

费劲千辛万苦，呕心沥血的大总结终于写好了。

 

主要包括了树状数组的三大应用以及详细的解释。

 

在此还要感谢http://www.cppblog.com/MatoNo1/archive/2011/03/19/142226.aspx的博文。

 

其实主要还是从中学习的，不过我想讲的更详细一点，更好理解一点。

 

期间还参考了好多博文，在此谢过~

 

阅读此文前请先阅读树状数组相关资料。

 

 

【应用一：改点求点】

 

这个是打酱油的应用。

 

完全不需要用到树状数组。

 

直接开个数组模拟就行了。

   

 

【应用二：改点求段】

 

也就是所谓求和。这也算是最基本的应用

 

可以参考：http://blog.csdn.net/Human_CK/archive/2011/06/13/6540532.aspx

               http://blog.csdn.net/Human_CK/archive/2011/06/12/6539722.aspx

 

三个函数在其中也是写的相当清楚。

 

求任意两点之间的和时也是用到了容斥原理。

 

这里要说明的一点是，“改点求段”的过程中，修改时是从左向右修改的（+lowbit(x)），求和时是从右向左求和的（-lowbit(x)）。

 

原因很简单，修改时总是修改跟其相关的上节点，求和时分为两部分，一部分是当前管辖的区域，另一部分是非管辖的区域。

 

在下面第三个应用中会详细说到。

 

 

【应用三：改段求点】

 

参考其应用（二维）：http://blog.csdn.net/Human_CK/archive/2011/06/14/6542909.aspx

 

后面的两个应用是重要且难以理解的。

 

如果数据规模比较小，可以直接用普通循环去修改，然后定义树状数组直接求和。

 

但是数据规模比较大时这个方法就太费劲了。

 

假设num数组用于存储原有数据。a数组用于树状数组的求和。

 

首先，需要一个辅助数组b。b存储的是一个改变量d，这个改变量表示num[1~i]的数都加上了d。

 

这样有什么好处呢？

 

如果要知道num[i]总共改变了多少，只需要知道num[i]之后的改变量的总和，即b[i~n]。也就是说，b[i]表示有对num[i]进行b[i]的改变（因为b[]的改变范围是1到n），b[i+1]表示有对num[i]进行b[i+1]的改变，……知道b[n]对num[i]进行b[n]的改变。

 

所以只要知道b[i~n]的值就可以算出对num[i]的总改变量。

 

那么这个b[i~n]的总和怎么求呢？

 

回到第二个应用，修改的时候是+lowbit，求和的时候是-lowbit。那么只要把它倒过来，修改的时候-lowbit，求和的时候-lowbit，就是我们所要求的结果。

 

可是为什么要这么做呢？

 

 

看这个图。我们平时计算用的都是这个图。这样的求法可以求出其左边的和。

 

反过来，节点1管辖1，节点2管辖（2,3），节点4管辖（4,5,6,7），其中节点6管辖（6,7），……如此这般。画个图就可以看明白的。

 

所以，倒过来求，就可以求出右边的和了。

 

到这里b[i~n]的和求出来了，表示为sum_B(i)，这个值就是num[i]的改变量。每次要求某个节点的值时就是num[i]+sum_B(i)。

  

 

【应用四：改段求段】

 

这里还需要另一个数组c。c表示从1到i的总改变量。

 

每次在b[i]上增加d时，只需要在c[i]上增加d*i，表示增加1到i个数同时增加d的总和。

 

那么此时的总和怎么算呢？

 

假设要求sum(i)（表示求1到i的总和)，那么b[i]之后的每个数值都表示对这个范围有改变，所以sum应该包括b[i~n]*i。

 

另外，i之前的所有数本来就有一个改变量为c，所以sum又得包括c[1~(i-1)]（这里只计算到i-1，因为i被前面的b[]算过了）。

 

所以这里的sum_B也是反向维护（这里的b[]跟应用三中的b一样），而c则属于正常维护。

 

需要注意当增量为0时不可以进行操作，否则会越界。

 

求某段和时也是使用了容斥原理。

 

 

【代码】

 

#include <iostream> using namespace std; const int maxn = 10000; int num[maxn+10]; //num if used to save the original numbers int a[maxn+10]; //a is used to save the sum of the array as a tree int b[maxn+10]; //b is used to save the changes between 1 and i int c[maxn+10]; //c is used to save the total change between 1 and i //calculate the bits behind the first 1(bit) counting from left ( including the 1 ) //the number 10110's lowbit is 10 which is the last two bits from 10110 inline int lowbit(int x) { return x & (-x); } //calculate the sum between 1 and x inline int pre_cal_sum(int *cal, int x) { int i, s = 0; for (i=x; i>0; i-=lowbit(i)) s += cal[i]; return s; } //calculate the sum between x and n inline int post_cal_sum(int *cal, int x, int n) { int i, s = 0; for (i=x; i<=n; i+=lowbit(i)) s += cal[i]; return s; } //update the nodes before x ( including x ) inline void pre_update(int *cal, int n, int x, int d, bool times) { int i; if (times) for (i=x; i>0; i-=lowbit(i)) cal[i] += x * d; else for (i=x; i>0; i-=lowbit(i)) cal[i] += d; } //update the nodes behind x ( including x ) inline void post_update(int *cal, int n, int x, int d, bool times) { int i; if (times) for (i=x; i<=n; i+=lowbit(i)) cal[i] += x * d; else for (i=x; i<=n; i+=lowbit(i)) cal[i] += d; } //get the sum of array A int GetSA(int *cal, int x) { return pre_cal_sum(cal, x); } //get the sum of array B int GetSB(int *cal, int x, int n) { return post_cal_sum(cal, x, n); } //get the sum of array C int GetSC(int *cal, int x) { return pre_cal_sum(cal, x-1); } //get the sum of a node after doing segment changing int sum_seg_to_nod(int x, int *src, int *b, int n) { return src[x] + GetSB(b, x, n); } //get the sum of a segment after doing segment changing int sum_seg_to_seg(int *a, int *b, int *c, int n, int x) { if (x) return GetSA(a,x)+GetSB(b, x, n)*x + GetSC(c, x); else return GetSA(a,x); } //update array A void UpA(int *cal, int n, int x, int d) { post_update(cal, n, x, d, false); } //update array B void UpB(int *cal, int n, int x, int d) { pre_update(cal, n, x, d, false); } //update array C void UpC(int *cal, int n, int x, int d) { post_update(cal, n, x, d, true); } //initialize array A void Init_A(int *src, int *dest, int n) { int i; for (i=1; i<=n; i++) dest[i] = 0; for (i=1; i<=n; i++) UpA(dest, n, i, src[i]); } //initialize array B void Init_B(int *dest, int n) { int i; for (i=1; i<=n; i++) dest[i] = 0; } //initialize array C void Init_C(int *dest, int n) { int i; for (i=1; i<=n; i++) dest[i] = 0; } //just for output void output(int *cal, int n) { int i; for (i=1; i<=n; i++) cout<<cal[i]<<' '; cout<<endl; } //test int main() { //x is the node to be updated //d is the number to be added to the node //s and t are the beginning and the ending of the segment //n is the the length of the array int i, s, t, x, d, n; s = 3; t = 7; n = 10; x = 6; d = 2; for (i=1; i<=10; i++) num[i] = i; //initialize array num cout<<"the original array num:/n"; output(num, n); //[node] to [segment] cout<<"/nnode to segment/n/n"; cout<<"the original array a:/n"; Init_A(num, a, n); output(a, n); cout<<"the original sum between "<<s<<" and "<<t<<" is : "<<GetSA(a, t)-GetSA(a, s-1)<<endl; cout<<"add "<<d<<" to num["<<x<<"]./n"; UpA(a, n, x, d); cout<<"the new array a:/n"; output(a, n); cout<<"the new sum between "<<s<<" and "<<t<<" is : "<<GetSA(a, t)-GetSA(a, s-1)<<endl; cout<<endl; //[segment] to [node] cout<<"/nsegment to node/n/n"; cout<<"the original array a:/n"; Init_A(num, a, n); output(a, n); cout<<"the original array b:/n"; Init_B(b, n); output(b, n); cout<<"the original num["<<x<<"]="<<sum_seg_to_nod(x, num, b, n)<<endl; cout<<"add "<<d<<" to numbers between "<<s<<" and "<<t<<"."<<endl; UpB(b, n, t, d); UpB(b, n, s-1, -d); cout<<"the new array b:/n"; output(b, n); cout<<"the new num["<<x<<"]="<<sum_seg_to_nod(x, num, b, n)<<endl; cout<<endl; //[segment] to [segment] cout<<"/nsegment to segment/n/n"; cout<<"the original array a:/n"; Init_A(num, a, n); output(a, n); cout<<"the original array b:/n"; Init_B(b, n); output(b, n); cout<<"the original array c:/n"; Init_C(c, n); output(c, n); cout<<"the original sum between "<<s+1<<" and "<<t+1<<" is : "<<sum_seg_to_seg(a,b,c,n,t+1)-sum_seg_to_seg(a,b,c,n,s)<<endl; cout<<"add "<<d<<" to numbers between "<<s<<" and "<<t<<"."<<endl; UpB(b, n, t, d); UpC(c, n, t, d); UpB(b, n, s-1, -d); UpC(c, n, s-1, -d); cout<<"the new array b:/n"; output(b, n); cout<<"the new array c:/n"; output(c, n); cout<<"the new sum between "<<s+1<<" and "<<t+1<<" is : "<<sum_seg_to_seg(a,b,c,n,t+1)-sum_seg_to_seg(a,b,c,n,s)<<endl; cout<<endl; return 0; }

 

 

【P.S】

 

上面这份代码也是写得挺烦的，主要是考虑了函数的精简。

 

今天研究了一整天，终于算是弄懂了。

 

好好休息一下~