序:说起树状数组,记得这是我正式去组里的第一天,一点基础都没有的我顿时就懵了,学长在上面讲,完全不知所云,于是那天晚上就失眠了。。。
正文:
一 树状数组就是给你一个数组int a[N],对这个数组进行如下操作:
(1)修改数组中某元素的值;
(2)查询该数组任意区间的和。
任何算法都是以减少复杂度为目的的。其复杂度为(M+Q)log2(N);
(M为修改次数,Q为查询次数,N为数组长度)
首先设一个C数组构成一个树状关系:C[x]分布在 log2(lowbit(x)) 层;每个C[k]只有一个父亲节点,子节点与其父节点之间的距离为lowbit(k)。
对(1)操作,修改某位置k的值,则对于数组C要修改包括a[k]的位置,k与下一个节点的距离为lowbit(k)。
对(2)操作,查询某位置k的sum(k),则一次查询包括k到1的sum,k与其下一个相关节点的位置的距离为lowbit(k)。
函数实现代码:
int lowbit(int x) //位运算求lowbit(x) { return x&(-x); } int getsum(int x) //求sum { int result=0; while(x>0) { result +=c[x]; x-=lowbit(x); } return result; } void add(int x) //修改位置为x的值,val为变化值,可为+-; { while(x<=Max) { c[x]+=val; x+=lowbit(x); } }
也可构造二维数状数组
int lowbit(int x) { return x&(-x); } void add(int x,int y,int val) { int i,j; for(i=x;i<=n;i+=lowbit(i)) for(j=y;j<=n;j+=lowbit(j)) c[i][j]+=val; } int getsum(int x,int y) { int i,j,sum=0; for(i=x;i>0;i-=lowbit(i)) for(j=y;j>0;j-=lowbit(j)) sum+=c[i][j]; return sum; }
二 应用
(1)假如给定一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[1],a[2]...a[i-1]中(即位 置i的左边)小于等于a[i]的数的个数。对此例b[] = {0,1,1,2,0}。 那么该如何去求得b[i]呢?
(2)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[1],a[2]...a[i-1]中(即位置i的左边)大于等于a[i]的数的个数。对此例b[] = {0,0,1,1,4}。 那么该如何去求得b[i]呢?
(3)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[i],a[i+1]...a[n]中(即位置i的右边)小于等于a[i]的数的个数。对此例b[] = {1,3,1,1,0}。 那么该如何去求得b[i]呢?
(4)假如给你一个数组a[ ] = {2,5,3,4,1},求b[i],b[i] 表示在a[i],a[i+1]...a[n]中(即位置i的右边)大于等于a[i]的数的个数。对此例b[] = {3,0,1,0,0}。 那么该如何去求得b[i]呢?
以上四种情况殊途同归,其中(2)可以将其编号倒过来用(1)的方法求;
(4)可以将其编号倒过来用(3)的方法求。
用数状数组函数调用方式
for(int i=1; i<=n; i++) { b[i] = getSum(a[i]); //求前a[i]项的和 update(a[i],1); //第a[i]个元素+1 }
虽然现在稍懂了一点,但是还是应用不好,不过吾将上下而求索。
原文:http://www.cnblogs.com/searchmushan/p/4396560.html