这两个函数是Binary index tree (Fenwick tree) 数据结构的修改实现。
这里有两张图片来补充 MikeCAT 的答案,显示 i 变量如何针对不同的值进行更新。
“get”函数:
为简化表示,假设函数输入的最大值为 15。
编号为t的节点代表树数组中的tree[t]。
如果您为 i 调用 get 函数,则返回值是 tree[i] 的总和加上所有 tree 的总和它们在数组中的索引是图片中 i 的父元素的数组元素,但零除外。
以下是一些示例:
get(15) = tree[15] + tree[14] + tree[12] + tree[8]
get(14) = tree[14] + tree[12] + tree[8]
get(13) = tree[13] + tree[12] + tree[8]
get(12) = tree[12] + tree[8]
get(11) = tree[11] + tree[10] + tree[8]
get(10) = tree[10] + tree[8]
get(9) = tree[9] + tree[8]
get(8) = tree[8]
get(7) = tree[7] + tree[6] + tree[4]
get(6) = tree[6] + tree[4]
get(5) = tree[5] + tree[4]
get(4) = tree[4]
get(3) = tree[3] + tree[2]
get(2) = tree[2]
上图中节点标签上的数字具有这样一个属性,即每个节点的父节点是该节点标签减去最不重要的一个 1(在@MikeCAT 答案上解释得很好)
“更新”功能:
为简单起见,假设 tree 数组的最大长度为 16。
update 功能有点复杂。
它将 val 添加到 tree[i] 和所有 tree 元素,它们的索引是带有标签 i 的节点的父节点> 在图片中。
update(16, val) --> tree[16] += val;
update(15, val) --> tree[15] += val, tree[16] += val;
update(14, val) --> tree[14] += val, tree[16] += val;
update(13, val) --> tree[13] += val, tree[14] += val; tree[16] += val;
update(12, val) --> tree[12] += val, tree[16] += val;
update(11, val) --> tree[11] += val, tree[12] += val, tree[16] += val;
update(10, val) --> tree[10] += val, tree[12] += val, tree[16] += val;
update(9, val) --> tree[9] += val, tree[10] += val, tree[12] += val, tree[16] += val;
update(8, val) --> tree[8] += val, tree[16] += val;
update(7, val) --> tree[7] += val, tree[8] += val, tree[16] += val;
update(6, val) --> tree[6] += val, tree[8] += val, tree[16] += val;
update(5, val) --> tree[5] += val, tree[6] += val, tree[8] += val, tree[16] += val;
update(4, val) --> tree[4] += val, tree[8] += val, tree[16] += val;
update(3, val) --> tree[3] += val, tree[4] += val, tree[8] += val, tree[16] += val;
update(2, val) --> tree[2] += val, tree[4] += val, tree[8] += val, tree[16] += val;
update(1, val) --> tree[1] += val, tree[2] += val, tree[4] += val, tree[8] += val, tree[16] += val;