树状数组
树状数组概述
树状数组是一个查询和修改复杂度都为log(n)的数据结构,假设数组a[1..n], 那么查询a[1]+...+a[n]的时间是log级别的,而且是一个在线的数据结构,支持随时修改某个元素的值,复杂度也为log级别。来观察这个图:点击打开链接令这棵树的结点编号为C1,C2...Cn。令每个结点的值为这棵树的值的总和,那么容易发现:C1 = A1C2 = A1 + A2C3 = A3C4 = A1 + A2 + A3 + A4C5 = A5C6 = A5 + A6C7 = A7C8 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8...C16 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8 + A9 + A10 + A11 + A12 + A13 + A14 + A15 + A16这里有一个有趣的性质:设节点编号为x,那么这个节点管辖的区间为2^k(其中k为x二进制末尾0的个数)个元素。因为这个区间最后一个元素必然为Ax,所以很明显:Cn = A(n – 2^k + 1) + ... + An算这个2^k有一个快捷的办法,定义一个函数如下即可:int lowbit(int x){return x&(x^(x–1));}利用机器补码特性,也可以写成:int lowbit(int x){return x&(-x);}当想要查询一个SUM(n)(求a[n]的和),可以依据如下算法即可:step1: 令sum = 0,转第二步;step2: 假如n <= 0,算法结束,返回sum值,否则sum = sum + Cn,转第三步;step3: 令n = n – lowbit(n),转第二步。
树状数组解法分析
void pus(int pat,int num)
{
while(pat<=N)
{
ans[pat]+=num;
pat+=lowbit(pat);
}
}int getsum(int pat)
{
int sum=0;
while(pat>=1)
{
sum+=ans[pat];
pat-=lowbit(pat);
}
return sum;
}