internal RbstNode <T> InsertRoot(RbstNode <T> rbstNode, T item)
{
if (rbstNode == null)
{
return(new RbstNode <T>(item));
}
if (Compare(rbstNode.Value, item) > 0)
{
rbstNode.Left = InsertRoot((RbstNode <T>)rbstNode.Left, item);
return(RotateToRight(rbstNode));
}
rbstNode.Right = InsertRoot((RbstNode <T>)rbstNode.Right, item);
return(RotateToLeft(rbstNode));
}
internal RbstNode <T> RotateToLeft(RbstNode <T> rbstNode)
{
var tmp = (RbstNode <T>)rbstNode.Right;
if (tmp == null || Compare(tmp.Value, rbstNode.Value) == 0)
{
return(rbstNode);
}
rbstNode.Right = tmp.Left;
tmp.Left = rbstNode;
FixSize(rbstNode);
FixSize(tmp);
return(tmp);
}
internal RbstNode <T> Join(RbstNode <T> first, RbstNode <T> second)
{
if (first == null)
{
return(second);
}
if (second == null)
{
return(first);
}
if (Random.Next(first.Size + second.Size) < first.Size)
{
first.Right = Join((RbstNode <T>)first.Right, second);
FixSize(first);
return(first);
}
second.Left = Join(first, (RbstNode <T>)second.Left);
FixSize(second);
return(second);
}
internal RbstNode <T> Insert(RbstNode <T> rbstNode, T item)
{
if (rbstNode == null)
{
return(new RbstNode <T>(item));
}
if (Random.Next(rbstNode.Size + 1) == 0)
{
return(InsertRoot(rbstNode, item));
}
if (Compare(rbstNode.Value, item) > 0)
{
rbstNode.Left = Insert((RbstNode <T>)rbstNode.Left, item);
}
else
{
rbstNode.Right = Insert((RbstNode <T>)rbstNode.Right, item);
}
FixSize(rbstNode);
return(rbstNode);
}
internal RbstNode <T> Remove(RbstNode <T> rbstNode, T item)
{
if (rbstNode == null)
{
return(null);
}
if (rbstNode.Value.Equals(item))
{
return(Join((RbstNode <T>)rbstNode.Left, (RbstNode <T>)rbstNode.Right));
}
if (Compare(rbstNode.Value, item) > 0)
{
rbstNode.Left = Remove((RbstNode <T>)rbstNode.Left, item);
}
else
{
rbstNode.Right = Remove((RbstNode <T>)rbstNode.Right, item);
}
FixSize(rbstNode);
return(rbstNode);
}
internal static void FixSize(RbstNode <T> rbstNode)
{
rbstNode.Size = GetSize((RbstNode <T>)rbstNode.Left) + GetSize((RbstNode <T>)rbstNode.Right) + 1;
}
internal static int GetSize(RbstNode <T> rbstNode) => rbstNode?.Size ?? 0;
