1.线段树
2.树状树组
<1>适合: 单点修改 ,区间查询问题
功能是线段树的子集,但是在 单点修改,区间查询上代码更简洁高效。
<2>实现思路:将[1…N]中元素分组管理,每个index末尾带0的索引都是小组领导,其中领导级别越大,末尾0越多。
求区间和问题中,对于树状树组中第i个元素 tr[i]就代表它带领的小组的和,则对于
单点修改,则修改tr[i]后,还需要把他的直属领导,以及直属领导领导都相应修改。问题是如何找到直属领导?
找到元素i的直属领导的方式,是在比i大的数中给末尾加0,可以通过i+lowbit(i)方式操作。
其中,lowbit(i)代表i的二进制中最右侧1的数字。例如
lowbit(1)=1
lowbit(2)=lowbit(10’b)=2
lowbit(3)=lowbit(11’b)=1
找上级领导过程就是下图绿色过程
int lowbit(int x){
return x&-x;
}
void add(int index,int val){
//末尾有0的是领导,所以向后找自己的领导就是往自己末尾添加0,即加lowbit
for(int i=index;i<=N;i+=lowbit(i)){
tr[i]+=val;
}
}
则查询区间和 range[left,right] 可以表示为 presum[right]-presum[left-1];
presum(i)为前缀和,即[1…i]的和。
对于tr[i],1…i的前缀和可以表示为tr[i] (i自己带领的小组和) 以及 找到小于i的各个其他小组的领导表示的和。找小于自己的各个小组领导的方式是 i-lowbit(i),也就是在自己小的数字中往末尾添0;即为下图的深蓝色过程。
//树状树组查询 1....index的前缀和
int presum(int index){
int sum=0;
//末尾有0的是领导,找比自己小的数中的领导,即在自己小的数字中,往末尾添0
for(int i=index;i>0;i-=lowbit(i)){
sum+=tr[i];
}
return sum;
}
则区间和操作为
int range(int left,int right){
return presum(right)-presum(left-1);
}
总结:
树状树组找直接领导,在数字增大方向往自己末尾添加0,==》i+lowbit(i)
树状树组找小于自己的其他小组领导(不是自己的小组成员,即不是自己的子节点) 在数字减小方向往自己末尾添加0,即 i-lowbit(i)。总之找领导就是要往末尾加0