题目大意:
一行N个方格,开始每个格子里的数都是0。现在动态地提出一些问题和修改:提问的形式是求某一个特定的子区间[a,b]中所有元素的和;修改的规则是指定某一个格子x,加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N≤1024,提问和修改的总数可能达到60000条。
题解:
为线段树每个节点增加一个Count域。表示所对应区间内元素之和。
每次修改一个格子,需要修改从叶结点到根结点路径上所有结点的值。
注意:题目中的区间是以元素为端点,因此[a,b]和[b,c]存在重合,这和我们之前讨论的区间定义不同。我们这里忽略预处理过程,直接使用之前的区间定义。
这一题不仅仅用线段树做,还可以用树状数组。
对于序列a,我们设一个数组C,定义
C[i] = a[i – 2^k + 1] + … + a[i],k为i在二进制下末尾0的个数。
代码:
if (l=a) and (r=b)
then count:=tree[p].count
else begin
m:=(l+r) div 2;
if b<=m
then count:=count(p*2,l,m,a,b)
else if a>=m then count:=count(p*2+1,m,r,a,b)
else count:=count(p*2,l,m,a,m)+count(p*2+1,m,r,m,b);
end
then count:=tree[p].count
else begin
m:=(l+r) div 2;
if b<=m
then count:=count(p*2,l,m,a,b)
else if a>=m then count:=count(p*2+1,m,r,a,b)
else count:=count(p*2,l,m,a,m)+count(p*2+1,m,r,m,b);
end