priority-queue-typed 2.0.5 → 2.1.0

package/src/data-structures/binary-tree/red-black-tree.ts

@@ -1,4 +1,20 @@
-import type { BinaryTreeDeleteResult, CRUD, EntryCallback, OptNode, RBTNColor, RedBlackTreeOptions } from '../../types';
+/**
+ * data-structure-typed
+ *
+ * @author Pablo Zeng
+ * @copyright Copyright (c) 2022 Pablo Zeng <zrwusa@gmail.com>
+ * @license MIT License
+ */
+
+import type {
+  BinaryTreeDeleteResult,
+  BinaryTreeOptions,
+  CRUD,
+  EntryCallback,
+  OptNode,
+  RBTNColor,
+  RedBlackTreeOptions
+} from '../../types';
 import { BST, BSTNode } from './bst';
 import { IBinaryTree } from '../../interfaces';
 
@@ -6,15 +22,14 @@ export class RedBlackTreeNode<K = any, V = any> extends BSTNode<K, V> {
   override parent?: RedBlackTreeNode<K, V> = undefined;
 
   /**
-   * The constructor initializes a node with a key, value, and color for a Red-Black Tree.
-   * @param {K} key - The `key` parameter is a key of type `K` that is used to identify the node in a
-   * Red-Black Tree data structure.
-   * @param {V} [value] - The `value` parameter in the constructor is an optional parameter of type
-   * `V`. It represents the value associated with the key in the data structure being constructed.
-   * @param {RBTNColor} [color=BLACK] - The `color` parameter in the constructor is used to specify the
-   * color of the node in a Red-Black Tree. It has a default value of 'BLACK' if not provided
-   * explicitly.
+   * Create a Red-Black Tree and optionally bulk-insert items.
+   * @remarks Time O(n log n), Space O(n)
+   * @param key - See parameter type for details.
+   * @param [value]- See parameter type for details.
+   * @param color - See parameter type for details.
+   * @returns New RedBlackTree instance.
    */
+
   constructor(key: K, value?: V, color: RBTNColor = 'BLACK') {
     super(key, value);
     this._color = color;
@@ -22,10 +37,23 @@ export class RedBlackTreeNode<K = any, V = any> extends BSTNode<K, V> {
 
   override _left?: RedBlackTreeNode<K, V> | null | undefined = undefined;
 
+  /**
+   * Get the left child pointer.
+   * @remarks Time O(1), Space O(1)
+   * @returns Left child node, or null/undefined.
+   */
+
   override get left(): RedBlackTreeNode<K, V> | null | undefined {
     return this._left;
   }
 
+  /**
+   * Set the left child and update its parent pointer.
+   * @remarks Time O(1), Space O(1)
+   * @param v - New left node, or null/undefined.
+   * @returns void
+   */
+
   override set left(v: RedBlackTreeNode<K, V> | null | undefined) {
     if (v) {
       v.parent = this;
@@ -35,10 +63,23 @@ export class RedBlackTreeNode<K = any, V = any> extends BSTNode<K, V> {
 
   override _right?: RedBlackTreeNode<K, V> | null | undefined = undefined;
 
+  /**
+   * Get the right child pointer.
+   * @remarks Time O(1), Space O(1)
+   * @returns Right child node, or null/undefined.
+   */
+
   override get right(): RedBlackTreeNode<K, V> | null | undefined {
     return this._right;
   }
 
+  /**
+   * Set the right child and update its parent pointer.
+   * @remarks Time O(1), Space O(1)
+   * @param v - New right node, or null/undefined.
+   * @returns void
+   */
+
   override set right(v: RedBlackTreeNode<K, V> | null | undefined) {
     if (v) {
       v.parent = this;
@@ -48,6 +89,11 @@ export class RedBlackTreeNode<K = any, V = any> extends BSTNode<K, V> {
 }
 
 /**
+ * RRRRed-Black Tree (self-balancing BST) supporting map-like mode and stable O(log n) updates.
+ * @remarks Time O(1), Space O(1)
+ * @template K
+ * @template V
+ * @template R
  * 1. Efficient self-balancing, but not completely balanced. Compared with AVLTree, the addition and deletion efficiency is high but the query efficiency is slightly lower.
  * 2. It is BST itself. Compared with Heap which is not completely ordered, RedBlackTree is completely ordered.
  * @example
@@ -94,21 +140,11 @@ export class RedBlackTreeNode<K = any, V = any> extends BSTNode<K, V> {
  *     );
  *     console.log(stocksInRange); // ['GOOGL', 'META', 'MSFT']
  */
-export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR = object>
-  extends BST<K, V, R, MK, MV, MR>
-  implements IBinaryTree<K, V, R, MK, MV, MR>
+
+export class RedBlackTree<K = any, V = any, R extends object = object>
+  extends BST<K, V, R>
+  implements IBinaryTree<K, V, R>
 {
-  /**
-   * This TypeScript constructor initializes a Red-Black Tree with optional keys, nodes, entries, or
-   * raw data.
-   * @param keysNodesEntriesOrRaws - The `keysNodesEntriesOrRaws` parameter in the constructor is an
-   * iterable that can contain either `K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined` objects or `R` objects. It
-   * is used to initialize the Red-Black Tree with keys, nodes, entries, or
-   * @param [options] - The `options` parameter in the constructor is of type `RedBlackTreeOptions<K,
-   * V, R>`. It is an optional parameter that allows you to specify additional options for the
-   * RedBlackTree class. These options could include configuration settings, behavior customization, or
-   * any other parameters that are specific to
-   */
   constructor(
     keysNodesEntriesOrRaws: Iterable<
       K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined | R
@@ -126,60 +162,34 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
 
   protected override _root: RedBlackTreeNode<K, V> | undefined;
 
+  /**
+   * Get the current root node.
+   * @remarks Time O(1), Space O(1)
+   * @returns Root node, or undefined.
+   */
   override get root(): RedBlackTreeNode<K, V> | undefined {
     return this._root;
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function creates a new Red-Black Tree node with the specified key, value, and color.
-   * @param {K} key - The key parameter represents the key value of the node being created. It is of
-   * type K, which is a generic type that can be replaced with any specific type when using the
-   * function.
-   * @param {V} [value] - The `value` parameter is an optional parameter that represents the value
-   * associated with the key in the node. It is not required and can be omitted if you only need to
-   * create a node with a key.
-   * @param {RBTNColor} [color=BLACK] - The "color" parameter is used to specify the color of the node
-   * in a Red-Black Tree. It can have two possible values: "RED" or "BLACK". By default, the color is
-   * set to "BLACK" if not specified.
-   * @returns A new instance of a RedBlackTreeNode with the specified key, value, and color is being
-   * returned.
+   * Create a red-black node for the given key/value (value ignored in map mode).
+   * @remarks Time O(1), Space O(1)
+   * @param key - See parameter type for details.
+   * @param [value] - See parameter type for details.
+   * @param color - See parameter type for details.
+   * @returns A new RedBlackTreeNode instance.
    */
-  override createNode(key: K, value?: V, color: RBTNColor = 'BLACK'): RedBlackTreeNode<K, V> {
+  override _createNode(key: K, value?: V, color: RBTNColor = 'BLACK'): RedBlackTreeNode<K, V> {
     return new RedBlackTreeNode<K, V>(key, this._isMapMode ? undefined : value, color);
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function creates a new Red-Black Tree with the specified options.
-   * @param [options] - The `options` parameter is an optional object that contains additional
-   * configuration options for creating the Red-Black Tree. It has the following properties:
-   * @returns a new instance of a RedBlackTree object.
+   * Type guard: check whether the input is a RedBlackTreeNode.
+   * @remarks Time O(1), Space O(1)
+   * @param keyNodeOrEntry - See parameter type for details.
+   * @returns True if the value is a RedBlackTreeNode.
    */
-  override createTree(options?: RedBlackTreeOptions<K, V, R>) {
-    return new RedBlackTree<K, V, R, MK, MV, MR>([], {
-      iterationType: this.iterationType,
-      isMapMode: this._isMapMode,
-      specifyComparable: this._specifyComparable,
-      toEntryFn: this._toEntryFn,
-      ...options
-    });
-  }
 
-  /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function checks if the input is an instance of the RedBlackTreeNode class.
-   * @param {K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined} keyNodeOrEntry - The parameter
-   * `keyNodeOrEntry` can be of type `R` or `K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined`.
-   * @returns a boolean value indicating whether the input parameter `keyNodeOrEntry` is
-   * an instance of the `RedBlackTreeNode` class.
-   */
   override isNode(
     keyNodeOrEntry: K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined
   ): keyNodeOrEntry is RedBlackTreeNode<K, V> {
@@ -187,31 +197,22 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The "clear" function sets the root node of a data structure to a sentinel value and resets the
-   * size counter to zero.
+   * Remove all nodes and clear the key→value store (if in map mode).
+   * @remarks Time O(n), Space O(1)
+   * @returns void
    */
+
   override clear() {
     super.clear();
     this._root = this.NIL;
   }
 
   /**
-   * Time Complexity: O(log n)
-   * Space Complexity: O(log n)
-   *
-   * The function adds a new node to a binary search tree and returns true if the node was successfully
-   * added.
-   * @param {K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined} keyNodeOrEntry - The parameter
-   * `keyNodeOrEntry` can accept a value of type `R` or `K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined`.
-   * @param {V} [value] - The `value` parameter is an optional value that you want to associate with
-   * the key in the data structure. It represents the value that you want to add or update in the data
-   * structure.
-   * @returns The method is returning a boolean value. If a new node is successfully added to the tree,
-   * the method returns true. If the node already exists and its value is updated, the method also
-   * returns true. If the node cannot be added or updated, the method returns false.
+   * Insert or replace an entry using BST order and red-black fix-up.
+   * @remarks Time O(log n), Space O(1)
+   * @param keyNodeOrEntry - Key, node, or [key, value] entry to insert.
+   * @param [value]- See parameter type for details.
+   * @returns True if inserted or updated; false if ignored.
    */
   override add(
     keyNodeOrEntry: K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined,
@@ -223,7 +224,6 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
     const insertStatus = this._insert(newNode);
 
     if (insertStatus === 'CREATED') {
-      // Ensure the root is black
       if (this.isRealNode(this._root)) {
         this._root.color = 'BLACK';
       } else {
@@ -241,19 +241,12 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(log n)
-   * Space Complexity: O(log n)
-   *
-   * The function overrides the delete method in a binary tree data structure to remove a node based on
-   * a given predicate and maintain the binary search tree properties.
-   * @param {K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined} keyNodeOrEntry - The `keyNodeOrEntry`
-   * parameter in the `override delete` method is used to specify the condition or key based on which a
-   * node should be deleted from the binary tree. It can be a key, a node, an entry, or a predicate
-   * function that determines which node(s) should be deleted.
-   * @returns The `override delete` method is returning an array of `BinaryTreeDeleteResult<RedBlackTreeNode<K, V>>`
-   * objects. Each object in the array contains information about the deleted node and whether
-   * balancing is needed.
+   * Delete a node by key/node/entry and rebalance as needed.
+   * @remarks Time O(log n), Space O(1)
+   * @param keyNodeOrEntry - Key, node, or [key, value] entry identifying the node to delete.
+   * @returns Array with deletion metadata (removed node, rebalancing hint if any).
    */
+
   override delete(
     keyNodeOrEntry: K | RedBlackTreeNode<K, V> | [K | null | undefined, V | undefined] | null | undefined
   ): BinaryTreeDeleteResult<RedBlackTreeNode<K, V>>[] {
@@ -310,7 +303,6 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
     if (this._isMapMode) this._store.delete(nodeToDelete.key);
     this._size--;
 
-    // If the original color was black, fix the tree
     if (originalColor === 'BLACK') {
       this._deleteFixup(replacementNode);
     }
@@ -321,60 +313,54 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(n)
-   * Space Complexity: O(n)
-   *
-   * The `map` function in TypeScript overrides the default behavior to create a new Red-Black Tree by
-   * applying a callback to each entry in the original tree.
-   * @param callback - A function that will be called for each entry in the tree, with parameters
-   * representing the key, value, index, and the tree itself. It should return an entry for the new
-   * tree.
-   * @param [options] - The `options` parameter in the `map` method is of type `RedBlackTreeOptions<MK, MV,
-   * MR>`. This parameter allows you to specify additional options or configurations for the Red-Black
-   * Tree that will be created during the mapping process. These options could include things like
-   * custom comparators
-   * @param {any} [thisArg] - The `thisArg` parameter in the `override map` function is used to specify
-   * the value of `this` when executing the `callback` function. It allows you to set the context
-   * (value of `this`) for the callback function. This can be useful when you want to access properties
-   * or
-   * @returns A new Red-Black Tree is being returned, where each entry has been transformed using the
-   * provided callback function.
+   * Transform entries into a like-kind red-black tree with possibly different key/value types.
+   * @remarks Time O(n), Space O(n)
+   * @template MK
+   * @template MV
+   * @template MR
+   * @param callback - Mapping function from (key, value, index, tree) to a new [key, value].
+   * @param [options] - See parameter type for details.
+   * @param [thisArg] - See parameter type for details.
+   * @returns A new RedBlackTree with mapped entries.
    */
-  override map(
+
+  override map<MK = K, MV = V, MR extends object = object>(
     callback: EntryCallback<K, V | undefined, [MK, MV]>,
-    options?: RedBlackTreeOptions<MK, MV, MR>,
-    thisArg?: any
+    options?: Partial<BinaryTreeOptions<MK, MV, MR>>,
+    thisArg?: unknown
   ): RedBlackTree<MK, MV, MR> {
-    const newTree = new RedBlackTree<MK, MV, MR>([], options);
+    const out = this._createLike<MK, MV, MR>([], options);
+
     let index = 0;
     for (const [key, value] of this) {
-      newTree.add(callback.call(thisArg, key, value, index++, this));
+      out.add(callback.call(thisArg, key, value, index++, this));
     }
-    return newTree;
+    return out;
   }
 
-  /**
-   * Time Complexity: O(n)
-   * Space Complexity: O(n)
-   *
-   * The function `clone` overrides the default cloning behavior to create a deep copy of a tree
-   * structure.
-   * @returns The `cloned` object is being returned.
-   */
-  override clone() {
-    const cloned = this.createTree();
-    this._clone(cloned);
-    return cloned;
+  protected override _createInstance<TK = K, TV = V, TR extends object = R>(
+    options?: Partial<RedBlackTreeOptions<TK, TV, TR>>
+  ): this {
+    const Ctor = this.constructor as unknown as new (
+      iter?: Iterable<TK | RedBlackTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
+      opts?: RedBlackTreeOptions<TK, TV, TR>
+    ) => RedBlackTree<TK, TV, TR>;
+    return new Ctor([], { ...this._snapshotOptions<TK, TV, TR>(), ...(options ?? {}) }) as unknown as this;
+  }
+
+  protected override _createLike<TK = K, TV = V, TR extends object = R>(
+    iter: Iterable<
+      TK | RedBlackTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR
+    > = [],
+    options?: Partial<RedBlackTreeOptions<TK, TV, TR>>
+  ): RedBlackTree<TK, TV, TR> {
+    const Ctor = this.constructor as unknown as new (
+      iter?: Iterable<TK | RedBlackTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
+      opts?: RedBlackTreeOptions<TK, TV, TR>
+    ) => RedBlackTree<TK, TV, TR>;
+    return new Ctor(iter, { ...this._snapshotOptions<TK, TV, TR>(), ...(options ?? {}) });
   }
 
-  /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function sets the root of a tree-like structure and updates the parent property of the new
-   * root.
-   * @param {RedBlackTreeNode<K, V> | undefined} v - v is a parameter of type RedBlackTreeNode<K, V> or undefined.
-   */
   protected override _setRoot(v: RedBlackTreeNode<K, V> | undefined) {
     if (v) {
       v.parent = undefined;
@@ -382,18 +368,6 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
     this._root = v;
   }
 
-  /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function replaces an old node with a new node while preserving the color of the old node.
-   * @param {RedBlackTreeNode<K, V>} oldNode - The `oldNode` parameter represents the node that needs to be replaced in
-   * the data structure.
-   * @param {RedBlackTreeNode<K, V>} newNode - The `newNode` parameter is of type `RedBlackTreeNode<K, V>`, which represents a node in a
-   * data structure.
-   * @returns The method is returning the result of calling the `_replaceNode` method from the
-   * superclass, with the `oldNode` and `newNode` parameters.
-   */
   protected override _replaceNode(
     oldNode: RedBlackTreeNode<K, V>,
     newNode: RedBlackTreeNode<K, V>
@@ -404,17 +378,12 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(log n)
-   * Space Complexity: O(log n)
-   *
-   * The `_insert` function inserts a node into a binary search tree and performs necessary fix-ups to
-   * maintain the red-black tree properties.
-   * @param {RedBlackTreeNode<K, V>} node - The `node` parameter represents the node that needs to be inserted into the
-   * binary search tree.
-   * @returns a string value indicating the result of the insertion operation. It can return either
-   * 'UPDATED' if the node with the same key already exists and was updated, or 'CREATED' if a new node
-   * was created and inserted into the tree.
+   * (Protected) Standard BST insert followed by red-black fix-up.
+   * @remarks Time O(log n), Space O(1)
+   * @param node - Node to insert.
+   * @returns Status string: 'CREATED' or 'UPDATED'.
    */
+
   protected _insert(node: RedBlackTreeNode<K, V>): CRUD {
     let current = this.root;
     let parent: RedBlackTreeNode<K, V> | undefined = undefined;
@@ -451,14 +420,13 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The function `_transplant` is used to replace a node `u` with another node `v` in a binary tree.
-   * @param {RedBlackTreeNode<K, V>} u - The parameter "u" represents a node in a binary tree.
-   * @param {RedBlackTreeNode<K, V> | undefined} v - The parameter `v` is of type `RedBlackTreeNode<K, V> | undefined`, which means it can
-   * either be a `RedBlackTreeNode<K, V>` object or `undefined`.
+   * (Protected) Transplant a subtree in place of another during deletion.
+   * @remarks Time O(1), Space O(1)
+   * @param u - Node to replace.
+   * @param v - Replacement subtree root (may be undefined).
+   * @returns void
    */
+
   protected _transplant(u: RedBlackTreeNode<K, V>, v: RedBlackTreeNode<K, V> | undefined): void {
     if (!u.parent) {
       this._setRoot(v);
@@ -474,37 +442,28 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(log n)
-   * Space Complexity: O(1)
-   *
-   * The `_insertFixup` function is used to fix the Red-Black Tree after inserting a new node.
-   * @param {RedBlackTreeNode<K, V> | undefined} z - The parameter `z` represents a node in the Red-Black Tree data
-   * structure. It can either be a valid node or `undefined`.
+   * (Protected) Restore red-black properties after insertion (recolor/rotate).
+   * @remarks Time O(log n), Space O(1)
+   * @param z - Recently inserted node.
+   * @returns void
    */
+
   protected _insertFixup(z: RedBlackTreeNode<K, V> | undefined): void {
-    // Continue fixing the tree as long as the parent of z is red
     while (z?.parent?.color === 'RED') {
-      // Check if the parent of z is the left child of its parent
       if (z.parent === z.parent.parent?.left) {
-        // Case 1: The uncle (y) of z is red
         const y = z.parent.parent.right;
         if (y?.color === 'RED') {
-          // Set colors to restore properties of Red-Black Tree
           z.parent.color = 'BLACK';
           y.color = 'BLACK';
           z.parent.parent.color = 'RED';
-          // Move up the tree to continue fixing
+
           z = z.parent.parent;
         } else {
-          // Case 2: The uncle (y) of z is black, and z is a right child
           if (z === z.parent.right) {
-            // Perform a left rotation to transform the case into Case 3
             z = z.parent;
             this._leftRotate(z);
           }
 
-          // Case 3: The uncle (y) of z is black, and z is a left child
-          // Adjust colors and perform a right rotation
           if (z && this.isRealNode(z.parent) && this.isRealNode(z.parent.parent)) {
             z.parent.color = 'BLACK';
             z.parent.parent.color = 'RED';
@@ -512,8 +471,6 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
           }
         }
       } else {
-        // Symmetric case for the right child (left and right exchanged)
-        // Follow the same logic as above with left and right exchanged
         const y: RedBlackTreeNode<K, V> | undefined = z?.parent?.parent?.left ?? undefined;
         if (y?.color === 'RED') {
           z.parent.color = 'BLACK';
@@ -535,26 +492,20 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
       }
     }
 
-    // Ensure that the root is black after fixing
     if (this.isRealNode(this._root)) this._root.color = 'BLACK';
   }
 
   /**
-   * Time Complexity: O(log n)
-   * Space Complexity: O(1)
-   *
-   * The `_deleteFixup` function is used to fix the red-black tree after a node deletion by adjusting
-   * the colors and performing rotations.
-   * @param {RedBlackTreeNode<K, V> | undefined} node - The `node` parameter represents a node in a binary tree. It can
-   * be either a valid node object or `undefined`.
-   * @returns The function does not return any value. It has a return type of `void`, which means it
-   * does not return anything.
+   * (Protected) Restore red-black properties after deletion (recolor/rotate).
+   * @remarks Time O(log n), Space O(1)
+   * @param node - Child that replaced the deleted node (may be undefined).
+   * @returns void
    */
+
   protected _deleteFixup(node: RedBlackTreeNode<K, V> | undefined): void {
-    // Early exit condition
     if (!node || node === this.root || node.color === 'BLACK') {
       if (node) {
-        node.color = 'BLACK'; // Ensure the final node is black
+        node.color = 'BLACK';
       }
       return;
     }
@@ -563,13 +514,12 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
       const parent: RedBlackTreeNode<K, V> | undefined = node.parent;
 
       if (!parent) {
-        break; // Ensure the loop terminates if there's an issue with the tree structure
+        break;
       }
 
       if (node === parent.left) {
         let sibling = parent.right;
 
-        // Cases 1 and 2: Sibling is red or both children of sibling are black
         if (sibling?.color === 'RED') {
           sibling.color = 'BLACK';
           parent.color = 'RED';
@@ -577,12 +527,10 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
           sibling = parent.right;
         }
 
-        // Case 3: Sibling's left child is black
         if ((sibling?.left?.color ?? 'BLACK') === 'BLACK') {
           if (sibling) sibling.color = 'RED';
           node = parent;
         } else {
-          // Case 4: Adjust colors and perform a right rotation
           if (sibling?.left) sibling.left.color = 'BLACK';
           if (sibling) sibling.color = parent.color;
           parent.color = 'BLACK';
@@ -590,10 +538,8 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
           node = this.root;
         }
       } else {
-        // Symmetric case for the right child (left and right exchanged)
         let sibling = parent.left;
 
-        // Cases 1 and 2: Sibling is red or both children of sibling are black
         if (sibling?.color === 'RED') {
           sibling.color = 'BLACK';
           if (parent) parent.color = 'RED';
@@ -601,12 +547,10 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
           if (parent) sibling = parent.left;
         }
 
-        // Case 3: Sibling's left child is black
         if ((sibling?.right?.color ?? 'BLACK') === 'BLACK') {
           if (sibling) sibling.color = 'RED';
           node = parent;
         } else {
-          // Case 4: Adjust colors and perform a left rotation
           if (sibling?.right) sibling.right.color = 'BLACK';
           if (sibling) sibling.color = parent.color;
           if (parent) parent.color = 'BLACK';
@@ -616,21 +560,18 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
       }
     }
 
-    // Ensure that the final node (possibly the root) is black
     if (node) {
       node.color = 'BLACK';
     }
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The `_leftRotate` function performs a left rotation on a given node in a binary tree.
-   * @param {RedBlackTreeNode<K, V> | undefined} x - The parameter `x` is of type `RedBlackTreeNode<K, V> | undefined`. It represents a
-   * node in a binary tree or `undefined` if there is no node.
-   * @returns void, which means it does not return any value.
+   * (Protected) Perform a left rotation around x.
+   * @remarks Time O(1), Space O(1)
+   * @param x - Pivot node to rotate around.
+   * @returns void
    */
+
   protected _leftRotate(x: RedBlackTreeNode<K, V> | undefined): void {
     if (!x || !x.right) {
       return;
@@ -658,14 +599,12 @@ export class RedBlackTree<K = any, V = any, R = object, MK = any, MV = any, MR =
   }
 
   /**
-   * Time Complexity: O(1)
-   * Space Complexity: O(1)
-   *
-   * The `_rightRotate` function performs a right rotation on a given node in a binary tree.
-   * @param {RedBlackTreeNode<K, V> | undefined} y - The parameter `y` is of type `RedBlackTreeNode<K, V> | undefined`. It represents a
-   * node in a binary tree or `undefined` if there is no node.
-   * @returns void, which means it does not return any value.
+   * (Protected) Perform a right rotation around y.
+   * @remarks Time O(1), Space O(1)
+   * @param y - Pivot node to rotate around.
+   * @returns void
    */
+
   protected _rightRotate(y: RedBlackTreeNode<K, V> | undefined): void {
     if (!y || !y.left) {
       return;