/// <summary>
/// do a right rotation
/// </summary>
private void rightRotate(BpTreeNode <T> leftSibling, BpTreeNode <T> rightSibling)
{
var parentIndex = getNextSeparatorIndex(leftSibling);
//move parent value to right
insertAt(rightSibling.Keys, 0, rightSibling.Parent.Keys[parentIndex]);
rightSibling.KeyCount++;
insertChild(rightSibling, 0, leftSibling.Children[leftSibling.KeyCount]);
if (rightSibling.Children[1] != null &&
rightSibling.Children[1].IsLeaf)
{
rightSibling.Keys[0] = rightSibling.Children[1].Keys[0];
}
//move rightmost element in left sibling to parent
rightSibling.Parent.Keys[parentIndex] = leftSibling.Keys[leftSibling.KeyCount - 1];
//remove rightmost element of left sibling
removeAt(leftSibling.Keys, leftSibling.KeyCount - 1);
leftSibling.KeyCount--;
removeChild(leftSibling, leftSibling.KeyCount + 1);
if (rightSibling.IsLeaf)
{
rightSibling.Keys[0] = rightSibling.Parent.Keys[parentIndex];
}
}
private void connectLeaves(BpTreeNode <T> node, BpTreeNode <T> left, BpTreeNode <T> right)
{
left.Next = right;
right.Prev = left;
if (node.Next != null)
{
right.Next = node.Next;
node.Next.Prev = right;
}
else
{
//bottom right most node
BottomRightNode = right;
}
if (node.Prev != null)
{
left.Prev = node.Prev;
node.Prev.Next = left;
}
else
{
//bottom left most node
BottomLeftNode = left;
}
}
/// <summary>
/// Find the leaf node to start initial insertion.
/// </summary>
private BpTreeNode <T> findInsertionLeaf(BpTreeNode <T> node, T newValue)
{
//if leaf then its time to insert
if (node.IsLeaf)
{
return(node);
}
//if not leaf then drill down to leaf
for (var i = 0; i < node.KeyCount; i++)
{
//current value is less than new value
//drill down to left child of current value
if (newValue.CompareTo(node.Keys[i]) < 0)
{
return(findInsertionLeaf(node.Children[i], newValue));
}
//current value is grearer than new value
//and current value is last element
if (node.KeyCount == i + 1)
{
return(findInsertionLeaf(node.Children[i + 1], newValue));
}
}
return(node);
}
/// <summary>
/// do a left rotation
/// </summary>
private void leftRotate(BpTreeNode <T> leftSibling, BpTreeNode <T> rightSibling)
{
var parentIndex = getNextSeparatorIndex(leftSibling);
//move root to left
leftSibling.Keys[leftSibling.KeyCount] = leftSibling.Parent.Keys[parentIndex];
leftSibling.KeyCount++;
setChild(leftSibling, leftSibling.KeyCount, rightSibling.Children[0]);
if (leftSibling.Children[leftSibling.KeyCount] != null &&
leftSibling.Children[leftSibling.KeyCount].IsLeaf)
{
leftSibling.Keys[leftSibling.KeyCount - 1] = leftSibling.Children[leftSibling.KeyCount].Keys[0];
}
//move right to parent
leftSibling.Parent.Keys[parentIndex] = rightSibling.Keys[0];
//remove right
removeAt(rightSibling.Keys, 0);
rightSibling.KeyCount--;
removeChild(rightSibling, 0);
if (rightSibling.IsLeaf)
{
rightSibling.Parent.Keys[parentIndex] = rightSibling.Keys[0];
}
if (leftSibling.IsLeaf && leftSibling.Parent.Children[0] != leftSibling)
{
parentIndex = getPrevSeparatorIndex(leftSibling);
leftSibling.Parent.Keys[parentIndex] = leftSibling.Keys[0];
}
}
public bool MoveNext()
{
if (current == null)
{
return(false);
}
if (asc)
{
if (index + 1 < current.KeyCount)
{
index++;
return(true);
}
}
else
{
if (index - 1 >= 0)
{
index--;
return(true);
}
}
current = asc ? current.Next : current.Prev;
var canMove = current != null && current.KeyCount > 0;
if (canMove)
{
index = asc ? 0 : current.KeyCount - 1;
}
return(canMove);
}
/// <summary>
/// optionally recursively update outdated index with new min of right node
/// after deletion of a value
/// </summary>
private void updateIndex(BpTreeNode <T> node, T deleteKey, bool spiralUp)
{
while (true)
{
if (node == null)
{
return;
}
if (node.IsLeaf || node.Children[0].IsLeaf)
{
node = node.Parent;
continue;
}
for (var i = 0; i < node.KeyCount; i++)
{
if (node.Keys[i].CompareTo(deleteKey) == 0)
{
node.Keys[i] = findMinNode(node.Children[i + 1]).Keys[0];
}
}
if (spiralUp)
{
node = node.Parent;
continue;
}
break;
}
}
internal BpTreeEnumerator(BpTree <T> tree, bool asc = true)
{
this.asc = asc;
startNode = asc ? tree.BottomLeftNode : tree.BottomRightNode;
current = startNode;
index = asc ? -1 : current.KeyCount;
}
/// <summary>
/// return the node containing max value which will be a leaf at the right most
/// </summary>
private BpTreeNode <T> findMaxNode(BpTreeNode <T> node)
{
while (true)
{
//if leaf return node
if (node.IsLeaf)
{
return(node);
}
node = node.Children[node.KeyCount];
}
}
private void setChild(BpTreeNode <T> parent, int childIndex, BpTreeNode <T> child)
{
parent.Children[childIndex] = child;
if (child == null)
{
return;
}
child.Parent = parent;
child.Index = childIndex;
}
private void removeChild(BpTreeNode <T> parent, int childIndex)
{
removeAt(parent.Children, childIndex);
//update indices
for (var i = childIndex; i <= parent.KeyCount; i++)
{
if (parent.Children[i] != null)
{
parent.Children[i].Index = i;
}
}
}
/// <summary>
/// Get next separator key of this child Node in parent
/// </summary>
private int getNextSeparatorIndex(BpTreeNode <T> node)
{
var parent = node.Parent;
if (node.Index == 0)
{
return(0);
}
if (node.Index == parent.KeyCount)
{
return(node.Index - 1);
}
return(node.Index);
}
/// <summary>
/// Insert to a node that is not full
/// </summary>
private void insertNonFullNode(ref BpTreeNode <T> node, T newValue,
BpTreeNode <T> newValueLeft, BpTreeNode <T> newValueRight)
{
var inserted = false;
//if left is not null
//then right should'nt be null
if (newValueLeft != null)
{
newValueLeft.Parent = node;
newValueRight.Parent = node;
}
//insert in sorted order
for (var i = 0; i < node.KeyCount; i++)
{
if (newValue.CompareTo(node.Keys[i]) >= 0)
{
continue;
}
insertAt(node.Keys, i, newValue);
node.KeyCount++;
//Insert children if any
setChild(node, i, newValueLeft);
insertChild(node, i + 1, newValueRight);
inserted = true;
break;
}
//newValue is the greatest
//element should be inserted at the end then
if (inserted)
{
return;
}
node.Keys[node.KeyCount] = newValue;
node.KeyCount++;
setChild(node, node.KeyCount - 1, newValueLeft);
setChild(node, node.KeyCount, newValueRight);
}
private void insertChild(BpTreeNode <T> parent, int childIndex, BpTreeNode <T> child)
{
insertAt(parent.Children, childIndex, child);
if (child != null)
{
child.Parent = parent;
}
//update indices
for (var i = childIndex; i <= parent.KeyCount; i++)
{
if (parent.Children[i] != null)
{
parent.Children[i].Index = i;
}
}
}
public bool MoveNext()
{
if (current == null)
{
return(false);
}
if (index + 1 < current.KeyCount)
{
index++;
return(true);
}
current = current.Next;
index = 0;
return(current != null && current.KeyCount > 0);
}
/// <summary>
/// Time complexity: O(log(n)).
/// </summary>
public void Insert(T newValue)
{
if (Root == null)
{
Root = new BpTreeNode <T>(maxKeysPerNode, null)
{
Keys = { [0] = newValue }
};
Root.KeyCount++;
Count++;
BottomLeftNode = Root;
return;
}
var leafToInsert = findInsertionLeaf(Root, newValue);
insertAndSplit(ref leafToInsert, newValue, null, null);
Count++;
}
/// <summary>
/// Balance a node which is short of Keys by rotations or merge
/// </summary>
private void balance(BpTreeNode <T> node, T deleteKey)
{
if (node == Root)
{
return;
}
if (node.KeyCount >= minKeysPerNode)
{
updateIndex(node, deleteKey, true);
return;
}
var rightSibling = getRightSibling(node);
if (rightSibling != null &&
rightSibling.KeyCount > minKeysPerNode)
{
leftRotate(node, rightSibling);
findMinNode(node);
updateIndex(node, deleteKey, true);
return;
}
var leftSibling = getLeftSibling(node);
if (leftSibling != null &&
leftSibling.KeyCount > minKeysPerNode)
{
rightRotate(leftSibling, node);
updateIndex(node, deleteKey, true);
return;
}
if (rightSibling != null)
{
sandwich(node, rightSibling, deleteKey);
}
else
{
sandwich(leftSibling, node, deleteKey);
}
}
private void mergeLeaves(BpTreeNode <T> newNode, BpTreeNode <T> leftSibling, BpTreeNode <T> rightSibling)
{
if (leftSibling.Prev != null)
{
newNode.Prev = leftSibling.Prev;
leftSibling.Prev.Next = newNode;
}
else
{
BottomLeftNode = newNode;
}
if (rightSibling.Next != null)
{
newNode.Next = rightSibling.Next;
rightSibling.Next.Prev = newNode;
}
else
{
BottomRightNode = newNode;
}
}
/// <summary>
/// Find the given value node under given node
/// </summary>
private BpTreeNode <T> find(BpTreeNode <T> node, T value)
{
//if leaf then its time to insert
if (node.IsLeaf)
{
for (var i = 0; i < node.KeyCount; i++)
{
if (comparer.Compare(value, node.Keys[i]) == 0)
{
return(node);
}
}
}
else
{
//if not leaf then drill down to leaf
for (var i = 0; i < node.KeyCount; i++)
{
//current value is less than new value
//drill down to left child of current value
if (comparer.Compare(value, node.Keys[i]) < 0)
{
return(find(node.Children[i], value));
}
//current value is grearer than new value
//and current value is last element
if (node.KeyCount == i + 1)
{
return(find(node.Children[i + 1], value));
}
}
}
return(null);
}
internal BpTreeEnumerator(BpTree <T> tree)
{
bottomLeftNode = tree.BottomLeftNode;
current = bottomLeftNode;
}
internal BpTreeNode(int maxKeysPerNode, BpTreeNode <T> parent)
: base(maxKeysPerNode)
{
Parent = parent;
Children = new BpTreeNode <T> [maxKeysPerNode + 1];
}
public void Reset()
{
current = startNode;
index = asc ? -1 : current.KeyCount;
}
public void Dispose()
{
current = null;
startNode = null;
}
/// <summary>
/// get left sibling node
/// </summary>
private BpTreeNode <T> getLeftSibling(BpTreeNode <T> node)
{
return(node.Index == 0 ? null : node.Parent.Children[node.Index - 1]);
}
/// <summary>
/// get the right sibling node
/// </summary>
private BpTreeNode <T> getRightSibling(BpTreeNode <T> node)
{
var parent = node.Parent;
return(node.Index == parent.KeyCount ? null : parent.Children[node.Index + 1]);
}
public void Reset()
{
current = bottomLeftNode;
index = -1;
}
public void Dispose()
{
current = null;
bottomLeftNode = null;
index = -1;
}
/// <summary>
/// Insert and split recursively up until no split is required
/// </summary>
private void insertAndSplit(ref BpTreeNode <T> node, T newValue,
BpTreeNode <T> newValueLeft, BpTreeNode <T> newValueRight)
{
//add new item to current node
//this increases the height of B+ tree by one by adding a new root at top
if (node == null)
{
node = new BpTreeNode <T>(maxKeysPerNode, null);
Root = node;
}
//if node is full
//then split node
//and insert new median to parent
if (node.KeyCount == maxKeysPerNode)
{
//divide the current node values + new Node as left & right sub nodes
var left = new BpTreeNode <T>(maxKeysPerNode, null);
var right = new BpTreeNode <T>(maxKeysPerNode, null);
//connect leaves via linked list for faster enumeration
if (node.IsLeaf)
{
left.Next = right;
right.Prev = left;
if (node.Next != null)
{
right.Next = node.Next;
node.Next.Prev = right;
}
if (node.Prev != null)
{
left.Prev = node.Prev;
node.Prev.Next = left;
}
else
{
//left most bottom node
BottomLeftNode = left;
}
}
//median of current Node
var currentMedianIndex = node.GetMedianIndex();
//init currentNode under consideration to left
var currentNode = left;
var currentNodeIndex = 0;
//new Median also takes new Value in to Account
var newMedian = default(T);
var newMedianSet = false;
var newValueInserted = false;
//keep track of each insertion
var insertionCount = 0;
//insert newValue and existing values in sorted order
//to left & right nodes
//set new median during sorting
for (var i = 0; i < node.KeyCount; i++)
{
//if insertion count reached new median
//set the new median by picking the next smallest value
if (!newMedianSet && insertionCount == currentMedianIndex)
{
newMedianSet = true;
//median can be the new value or node.keys[i] (next node key)
//whichever is smaller
if (!newValueInserted && newValue.CompareTo(node.Keys[i]) < 0)
{
//median is new value
newMedian = newValue;
newValueInserted = true;
if (newValueLeft != null)
{
setChild(currentNode, currentNode.KeyCount, newValueLeft);
}
//now fill right node
currentNode = right;
currentNodeIndex = 0;
if (newValueRight != null)
{
setChild(currentNode, 0, newValueRight);
}
i--;
insertionCount++;
continue;
}
//median is next node
newMedian = node.Keys[i];
//now fill right node
currentNode = right;
currentNodeIndex = 0;
continue;
}
//pick the smaller among newValue & node.Keys[i]
//and insert in to currentNode (left & right nodes)
//if new Value was already inserted then just copy from node.Keys in sequence
//since node.Keys is already in sorted order it should be fine
if (newValueInserted || node.Keys[i].CompareTo(newValue) < 0)
{
currentNode.Keys[currentNodeIndex] = node.Keys[i];
currentNode.KeyCount++;
//if child is set don't set again
//the child was already set by last newValueRight or last node
if (currentNode.Children[currentNodeIndex] == null)
{
setChild(currentNode, currentNodeIndex, node.Children[i]);
}
setChild(currentNode, currentNodeIndex + 1, node.Children[i + 1]);
}
else
{
currentNode.Keys[currentNodeIndex] = newValue;
currentNode.KeyCount++;
setChild(currentNode, currentNodeIndex, newValueLeft);
setChild(currentNode, currentNodeIndex + 1, newValueRight);
i--;
newValueInserted = true;
}
currentNodeIndex++;
insertionCount++;
}
//could be that thew newKey is the greatest
//so insert at end
if (!newValueInserted)
{
currentNode.Keys[currentNodeIndex] = newValue;
currentNode.KeyCount++;
setChild(currentNode, currentNodeIndex, newValueLeft);
setChild(currentNode, currentNodeIndex + 1, newValueRight);
}
if (node.IsLeaf)
{
insertAt(right.Keys, 0, newMedian);
right.KeyCount++;
}
//insert overflow element (newMedian) to parent
var parent = node.Parent;
insertAndSplit(ref parent, newMedian, left, right);
}
//newValue have room to fit in this node
//so just insert in right spot in asc order of keys
else
{
insertNonFullNode(ref node, newValue, newValueLeft, newValueRight);
}
}
/// <summary>
/// merge two adjacent siblings to one node
/// </summary>
private void sandwich(BpTreeNode <T> leftSibling, BpTreeNode <T> rightSibling, T deleteKey)
{
var separatorIndex = getNextSeparatorIndex(leftSibling);
var parent = leftSibling.Parent;
var newNode = new BpTreeNode <T>(maxKeysPerNode, leftSibling.Parent);
//if leaves are merged then update the Next & Prev pointers
if (leftSibling.IsLeaf)
{
if (leftSibling.Prev != null)
{
newNode.Prev = leftSibling.Prev;
leftSibling.Prev.Next = newNode;
}
else
{
BottomLeftNode = newNode;
}
if (rightSibling.Next != null)
{
newNode.Next = rightSibling.Next;
rightSibling.Next.Prev = newNode;
}
}
var newIndex = 0;
for (var i = 0; i < leftSibling.KeyCount; i++)
{
newNode.Keys[newIndex] = leftSibling.Keys[i];
if (leftSibling.Children[i] != null)
{
setChild(newNode, newIndex, leftSibling.Children[i]);
}
if (leftSibling.Children[i + 1] != null)
{
setChild(newNode, newIndex + 1, leftSibling.Children[i + 1]);
}
newIndex++;
}
//special case when left sibling is empty
if (leftSibling.KeyCount == 0 && leftSibling.Children[0] != null)
{
setChild(newNode, newIndex, leftSibling.Children[0]);
}
if (!rightSibling.IsLeaf)
{
newNode.Keys[newIndex] = parent.Keys[separatorIndex];
newIndex++;
}
for (var i = 0; i < rightSibling.KeyCount; i++)
{
newNode.Keys[newIndex] = rightSibling.Keys[i];
if (rightSibling.Children[i] != null)
{
setChild(newNode, newIndex, rightSibling.Children[i]);
//when a left leaf is added as right leaf
//we need to push parent key as first node of new leaf
if (i == 0 && rightSibling.Children[i].IsLeaf &&
rightSibling.Children[i].Keys[0].CompareTo(newNode.Keys[newIndex - 1]) != 0)
{
insertAt(rightSibling.Children[i].Keys, 0, newNode.Keys[newIndex - 1]);
rightSibling.Children[i].KeyCount++;
}
}
if (rightSibling.Children[i + 1] != null)
{
setChild(newNode, newIndex + 1, rightSibling.Children[i + 1]);
}
newIndex++;
}
//special case when left sibling is empty
if (rightSibling.KeyCount == 0 && rightSibling.Children[0] != null)
{
setChild(newNode, newIndex, rightSibling.Children[0]);
if (newNode.Children[newIndex].IsLeaf)
{
newNode.Keys[newIndex - 1] = newNode.Children[newIndex].Keys[0];
}
}
newNode.KeyCount = newIndex;
setChild(parent, separatorIndex, newNode);
removeAt(parent.Keys, separatorIndex);
parent.KeyCount--;
removeChild(parent, separatorIndex + 1);
if (newNode.IsLeaf && newNode.Parent.Children[0] != newNode)
{
separatorIndex = getPrevSeparatorIndex(newNode);
newNode.Parent.Keys[separatorIndex] = newNode.Keys[0];
}
updateIndex(newNode, deleteKey, false);
if (parent.KeyCount == 0 &&
parent == Root)
{
Root = newNode;
Root.Parent = null;
if (Root.KeyCount == 0)
{
Root = null;
}
return;
}
if (parent.KeyCount < minKeysPerNode)
{
balance(parent, deleteKey);
}
updateIndex(newNode, deleteKey, true);
}
