1.lowbit操作
概念
一种位运算操作,用于计算出数字的二进制表达中的最低位的1以及后面所有的0。例如lowbit(4)=4
lowbit(6)=2
。
代码实现
代码原理是利用计算机存储特性,不用深究原理,只需记住代码。时间复杂度:O(1)
int lowbit(int x)
{
return x&-x;
}
2.树状数组概念
树状数组用来存储一段区间的和,例如tree[i]
的区间定义为:存储以i
结尾,区间长度为lowbit(i)
的区间和。即对与i其管辖的区间(即区间和的范围)为[i-lowbit(i)+1, i]
。例如tree[4]=a[1]+a[2]+a[3]+a[4]
,tree[6]=a[5]+a[6]
3.重要函数实现
3.1给原数组a[k]加上x
如给a[3]加上x,对应tree数组要修改的有t[3]、t[4]、t[8]、t[16]…
即对3、3+lowbit(3)=4、4+lowbit(4)=8…从3开始去找接下来的t是进行加等于lowbit操作,直到范围超过n。
代码实现
void update(int k,int x)
{
for(int i = k; i <=n; i += lowbit(i))
{
tree[i] += k;
}
}
3.2返回区间[1,k]的和
相应的减等于lowbit并求和,直到减到0为止,可以先建立树状数组在画图理解。
代码实现
int getsum(int k)
{
int sum = 0;
for(int i = k; i > 0; --i)
{
sum += t[i];
}
return sum;
}
(注:i的终止条件是i > 0)