在每个键上,您应该存储额外的数据来记录该节点下有多少键(包括节点本身)。
为了保持这一点,insert(k) 函数将不得不遍历新键 k 的所有祖先,并增加它们的值。这将使插入 O(log n) + O(log n),仍然是 O(log n),因此不会影响复杂性。delete(k) 必须做同样的事情,除了减少值。平衡操作也必须考虑到这一点。
然后,order(k) 将沿着树向下移动到 k:每次它移动到一个节点时,它应该将左侧有多少键的计数加到总数中,并返回这个总和。
编辑:我改变了节点和键之间“节点”的歧义,因为它们在 B 树中是不同的(一个节点可以包含多个键)。然而,该算法应该推广到大多数树数据结构。
这是B树的算法:
#In python-ish (untested psuedocode)
#root is the root of the tree
#Each node is expected to have an array named "keys",
# which contains the keys in the node.
#Each node is expected to have an array named "child_nodes",
# which contains the children of the node, if the node has children.
#If a node has children, this should be true: len(child_nodes) == len(keys) + 1
def inorder(q):
order_count = 0
current_node = root
while True:
#if q is after all keys in the node, then we will go to the last child node
next_child_node_i = len(current_node.keys)
#now see if q is in between any of the nodes
#for each key-index in the keys array (ie. if the node contains 3 keys,
# keyi will be in range [0-2] .)
for keyi in range(len(current_node.keys)):
#retrieve the value of the key, so we can do comparison
current_key = current_node.keys[keyi]
if current_key < q:
#We are trying to find which child node to go down to next,
# for now we will choose the child directly to the left of this key,
#But we continue to look through the rest of the keys, to find which
# two keys q lies in between.
#before we continue, we should count this key in the order too:
#if this is not a leaf node,
if len(current_node.children) != 0:
#retrieve the the recorded child count of the sub-tree
order_count += current_node.children[keyi].recorded_descendant_key_count
#add one for the key in this node that we are skipping.
order_count += 1
continue
if q < current_key:
#We found a key in the current node that is greater than q.
#Thus we continue to the next level between this and the previous key.
next_child_node_i = keyi
break
#we finally found q,
if q == current_key:
#now we just return the count
return order_count
#once we are here, we know which keys q lies between
# (or if it belongs at the beginning or end), and thus which child to travel down to.
#If this is a leaf node (it has no children),
# then q was not found.
if len(current_node.child_nodes) == 0:
#Possible behaviors: throw exception, or just return the place in the order
# where q *would* go, like so:
return order
#Travel down a level
current_node = current_node.child_nodes[next_child_node_i]