avl-tree-typed 1.47.5 → 1.47.7

package/src/data-structures/binary-tree/rb-tree.ts

@@ -8,8 +8,10 @@
 
 import {
   BiTreeDeleteResult,
+  BSTNodeKeyOrNode,
   BTNCallback,
   BTNKey,
+  BTNodeExemplar,
   IterationType,
   RBTNColor,
   RBTreeOptions,
@@ -43,21 +45,23 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
   extends BST<V, N, TREE>
   implements IBinaryTree<V, N, TREE> {
   Sentinel: N = new RedBlackTreeNode<V>(NaN) as unknown as N;
-  override options: RBTreeOptions;
 
   /**
-   * The constructor function initializes a Red-Black Tree with an optional set of options.
-   * @param {RBTreeOptions} [options] - The `options` parameter is an optional object that can be
-   * passed to the constructor. It is used to configure the RBTree object with specific options.
+   * This is the constructor function for a Red-Black Tree data structure in TypeScript, which
+   * initializes the tree with optional elements and options.
+   * @param [elements] - The `elements` parameter is an optional iterable of `BTNodeExemplar<V, N>`
+   * objects. It represents the initial elements that will be added to the RBTree during its
+   * construction. If this parameter is provided, the `addMany` method is called to add all the
+   * elements to the
+   * @param [options] - The `options` parameter is an optional object that allows you to customize the
+   * behavior of the RBTree. It is of type `Partial<RBTreeOptions>`, which means that you can provide
+   * only a subset of the properties defined in the `RBTreeOptions` interface.
    */
-  constructor(options?: RBTreeOptions) {
-    super(options);
-    if (options) {
-      this.options = { iterationType: IterationType.ITERATIVE, comparator: (a, b) => a - b, ...options }
-    } else {
-      this.options = { iterationType: IterationType.ITERATIVE, comparator: (a, b) => a - b };
-    }
+  constructor(elements?: Iterable<BTNodeExemplar<V, N>>, options?: Partial<RBTreeOptions>) {
+    super([], options);
+
     this._root = this.Sentinel;
+    if (elements) super.addMany(elements);
   }
 
   protected _root: N;
@@ -72,12 +76,34 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
     return this._size;
   }
 
+  /**
+   * The function creates a new Red-Black Tree node with the specified key, value, and color.
+   * @param {BTNKey} key - The key parameter is the key value associated with the node. It is used to
+   * identify and compare nodes in the Red-Black Tree.
+   * @param {V} [value] - The `value` parameter is an optional parameter that represents the value
+   * associated with the node. It is of type `V`, which is a generic type that can be replaced with any
+   * specific type when using the `createNode` method.
+   * @param {RBTNColor} color - The "color" parameter is used to specify the color of the node in a
+   * Red-Black Tree. It can be either "RED" or "BLACK". By default, the color is set to "BLACK".
+   * @returns The method is returning a new instance of a RedBlackTreeNode with the specified key,
+   * value, and color.
+   */
   override createNode(key: BTNKey, value?: V, color: RBTNColor = RBTNColor.BLACK): N {
     return new RedBlackTreeNode<V, N>(key, value, color) as N;
   }
 
+  /**
+   * The function creates a Red-Black Tree with the specified options and returns it.
+   * @param {RBTreeOptions} [options] - The `options` parameter is an optional object that can be
+   * passed to the `createTree` function. It is used to customize the behavior of the `RedBlackTree`
+   * class.
+   * @returns a new instance of a RedBlackTree object.
+   */
   override createTree(options?: RBTreeOptions): TREE {
-    return new RedBlackTree<V, N, TREE>({ ...this.options, ...options }) as TREE;
+    return new RedBlackTree<V, N, TREE>([], {
+      iterationType: this.iterationType,
+      comparator: this.comparator, ...options
+    }) as TREE;
   }
 
   /**
@@ -85,27 +111,28 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
    * Space Complexity: O(1)
    */
 
+
   /**
-   * Time Complexity: O(log n) on average (where n is the number of nodes in the tree)
-   * Space Complexity: O(1)
-   *
-   * The `add` function adds a new node to a Red-Black Tree data structure.
-   * @param {BTNKey | N | null | undefined} keyOrNode - The `keyOrNode` parameter can be one of the
-   * following types:
-   * @param {V} [value] - The `value` parameter is an optional value that can be associated with the
-   * key in the node being added to the Red-Black Tree.
-   * @returns The method returns either a node (`N`) or `undefined`.
+   * The function adds a node to a Red-Black Tree data structure.
+   * @param keyOrNodeOrEntry - The `keyOrNodeOrEntry` parameter can be one of the following:
+   * @returns The method `add` returns either an instance of `N` (the node that was added) or
+   * `undefined`.
    */
-  override add(keyOrNode: BTNKey | N | null | undefined, value?: V): N | undefined {
+  override add(keyOrNodeOrEntry: BTNodeExemplar<V, N>): N | undefined {
     let node: N;
-    if (this.isNodeKey(keyOrNode)) {
-      node = this.createNode(keyOrNode, value, RBTNColor.RED);
-    } else if (keyOrNode instanceof RedBlackTreeNode) {
-      node = keyOrNode;
-    } else if (keyOrNode === null) {
-      return;
-    } else if (keyOrNode === undefined) {
+    if (this.isNodeKey(keyOrNodeOrEntry)) {
+      node = this.createNode(keyOrNodeOrEntry, undefined, RBTNColor.RED);
+    } else if (keyOrNodeOrEntry instanceof RedBlackTreeNode) {
+      node = keyOrNodeOrEntry;
+    } else if (keyOrNodeOrEntry === null || keyOrNodeOrEntry === undefined) {
       return;
+    } else if (this.isEntry(keyOrNodeOrEntry)) {
+      const [key, value] = keyOrNodeOrEntry;
+      if (key === undefined || key === null) {
+        return;
+      } else {
+        node = this.createNode(key, value, RBTNColor.RED);
+      }
     } else {
       return;
     }
@@ -124,6 +151,9 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
         } else if (node.key > x.key) {
           x = x?.right;
         } else {
+          if (node !== x) {
+            this._replaceNode(x, node)
+          }
           return;
         }
       }
@@ -235,6 +265,11 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
     return ans;
   }
 
+  /**
+   * Time Complexity: O(log n) on average (where n is the number of nodes in the tree)
+   * Space Complexity: O(1)
+   */
+
   override isRealNode(node: N | undefined): node is N {
     return node !== this.Sentinel && node !== undefined;
   }
@@ -289,11 +324,11 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
   getNode<C extends BTNCallback<N>>(
     identifier: ReturnType<C> | undefined,
     callback: C = this._defaultOneParamCallback as C,
-    beginRoot: BTNKey | N | undefined = this.root,
-    iterationType = this.options.iterationType
+    beginRoot: BSTNodeKeyOrNode<N> = this.root,
+    iterationType = this.iterationType
   ): N | null | undefined {
     if ((identifier as any) instanceof BinaryTreeNode) callback = (node => node) as C;
-    beginRoot = this.ensureNotKey(beginRoot);
+    beginRoot = this.ensureNode(beginRoot);
     return this.getNodes(identifier, callback, true, beginRoot, iterationType)[0] ?? undefined;
   }
 
@@ -351,6 +386,11 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
     return y!;
   }
 
+  /**
+   * Time Complexity: O(log n) on average (where n is the number of nodes in the tree)
+   * Space Complexity: O(1)
+   */
+
   override clear() {
     this._root = this.Sentinel;
     this._size = 0;
@@ -582,4 +622,19 @@ export class RedBlackTree<V = any, N extends RedBlackTreeNode<V, N> = RedBlackTr
     }
     this.root.color = RBTNColor.BLACK;
   }
+
+  /**
+   * The function replaces an old node with a new node while preserving the color of the old node.
+   * @param {N} oldNode - The `oldNode` parameter represents the node that needs to be replaced in a
+   * data structure. It is of type `N`, which is the type of the nodes in the data structure.
+   * @param {N} newNode - The `newNode` parameter is the node that will replace the `oldNode` in the
+   * data structure.
+   * @returns The method is returning the result of calling the `_replaceNode` method from the
+   * superclass, passing in the `oldNode` and `newNode` as arguments.
+   */
+  protected _replaceNode(oldNode: N, newNode: N): N {
+    newNode.color = oldNode.color;
+
+    return super._replaceNode(oldNode, newNode)
+  }
 }

package/src/data-structures/binary-tree/tree-multimap.ts

@@ -5,8 +5,14 @@
  * @copyright Copyright (c) 2022 Tyler Zeng <zrwusa@gmail.com>
  * @license MIT License
  */
-import type { BTNKey, TreeMultimapNodeNested, TreeMultimapOptions } from '../../types';
-import { BiTreeDeleteResult, BTNCallback, CP, FamilyPosition, IterationType, TreeMultimapNested } from '../../types';
+import type {
+  BSTNodeKeyOrNode,
+  BTNKey,
+  BTNodeExemplar,
+  TreeMultimapNodeNested,
+  TreeMultimapOptions
+} from '../../types';
+import { BiTreeDeleteResult, BTNCallback, FamilyPosition, IterationType, TreeMultimapNested } from '../../types';
 import { IBinaryTree } from '../../interfaces';
 import { AVLTree, AVLTreeNode } from './avl-tree';
 
@@ -40,27 +46,18 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
   extends AVLTree<V, N, TREE>
   implements IBinaryTree<V, N, TREE> {
 
-  override options: TreeMultimapOptions;
-
-  /**
-   * The constructor function for a TreeMultimap class in TypeScript, which extends another class and sets an option to
-   * merge duplicated values.
-   * @param {TreeMultimapOptions} [options] - An optional object that contains additional configuration options for the
-   * TreeMultimap.
-   */
-  constructor(options: TreeMultimapOptions = { iterationType: IterationType.ITERATIVE }) {
-    super(options);
-    if (options) {
-      this.options = { iterationType: IterationType.ITERATIVE, comparator: (a, b) => a - b, ...options }
-    } else {
-      this.options = { iterationType: IterationType.ITERATIVE, comparator: (a, b) => a - b };
-    }
+  constructor(elements?: Iterable<BTNodeExemplar<V, N>>, options?: Partial<TreeMultimapOptions>) {
+    super([], options);
+    if (elements) this.addMany(elements);
   }
 
   private _count = 0;
 
+  // TODO the _count is not accurate after nodes count modified
   get count(): number {
-    return this._count;
+    let sum = 0;
+    this.subTreeTraverse(node => sum += node.count);
+    return sum;
   }
 
   /**
@@ -77,7 +74,10 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
   }
 
   override createTree(options?: TreeMultimapOptions): TREE {
-    return new TreeMultimap<V, N, TREE>({ ...this.options, ...options }) as TREE;
+    return new TreeMultimap<V, N, TREE>([], {
+      iterationType: this.iterationType,
+      comparator: this.comparator, ...options
+    }) as TREE;
   }
 
   /**
@@ -89,127 +89,62 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
    * Time Complexity: O(log n) - logarithmic time, where "n" is the number of nodes in the tree. The add method of the superclass (AVLTree) has logarithmic time complexity.
    * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
    *
-   * The `add` function adds a new node to the tree multimap, updating the count if the key already
-   * exists, and balances the tree if necessary.
-   * @param {BTNKey | N | null | undefined} keyOrNode - The `keyOrNode` parameter can be one of the
-   * following types:
-   * @param {V} [value] - The `value` parameter represents the value associated with the key that is
-   * being added to the tree. It is an optional parameter, so it can be omitted if not needed.
+   * The `add` function overrides the base class `add` function to add a new node to the tree multimap
+   * and update the count.
+   * @param keyOrNodeOrEntry - The `keyOrNodeOrEntry` parameter can be one of the following:
    * @param [count=1] - The `count` parameter is an optional parameter that specifies the number of
-   * times the key-value pair should be added to the multimap. If not provided, the default value is 1.
-   * @returns a node (`N`) or `undefined`.
+   * times the key or node or entry should be added to the multimap. If not provided, the default value
+   * is 1.
+   * @returns either a node (`N`) or `undefined`.
    */
-  override add(keyOrNode: BTNKey | N | null | undefined, value?: V, count = 1): N | undefined {
-    if (keyOrNode === null) return undefined;
-    let inserted: N | undefined = undefined,
-      newNode: N | undefined;
-    if (keyOrNode instanceof TreeMultimapNode) {
-      newNode = this.createNode(keyOrNode.key, keyOrNode.value, keyOrNode.count);
-    } else if (keyOrNode === undefined) {
-      newNode = undefined;
+  override add(keyOrNodeOrEntry: BTNodeExemplar<V, N>, count = 1): N | undefined {
+    let newNode: N | undefined;
+    if (keyOrNodeOrEntry === undefined || keyOrNodeOrEntry === null) {
+      return;
+    } else if (keyOrNodeOrEntry instanceof TreeMultimapNode) {
+      newNode = keyOrNodeOrEntry;
+    } else if (this.isNodeKey(keyOrNodeOrEntry)) {
+      newNode = this.createNode(keyOrNodeOrEntry, undefined, count);
+    } else if (this.isEntry(keyOrNodeOrEntry)) {
+      const [key, value] = keyOrNodeOrEntry;
+      if (key === undefined || key === null) {
+        return;
+      } else {
+        newNode = this.createNode(key, value, count);
+      }
     } else {
-      newNode = this.createNode(keyOrNode, value, count);
+      return;
     }
-    if (!this.root) {
-      this._setRoot(newNode);
-      this._size = this.size + 1;
-      if (newNode) this._count += newNode.count;
-      inserted = this.root;
-    } else {
-      let cur = this.root;
-      let traversing = true;
-      while (traversing) {
-        if (cur) {
-          if (newNode) {
-            if (this._compare(cur.key, newNode.key) === CP.eq) {
-              cur.value = newNode.value;
-              cur.count += newNode.count;
-              this._count += newNode.count;
-              traversing = false;
-              inserted = cur;
-            } else if (this._compare(cur.key, newNode.key) === CP.gt) {
-              // Traverse left of the node
-              if (cur.left === undefined) {
-                //Add to the left of the current node
-                cur.left = newNode;
-                this._size = this.size + 1;
-                this._count += newNode.count;
-
-                traversing = false;
-                inserted = cur.left;
-              } else {
-                //Traverse the left of the current node
-                if (cur.left) cur = cur.left;
-              }
-            } else if (this._compare(cur.key, newNode.key) === CP.lt) {
-              // Traverse right of the node
-              if (cur.right === undefined) {
-                //Add to the right of the current node
-                cur.right = newNode;
-                this._size = this.size + 1;
-                this._count += newNode.count;
-
-                traversing = false;
-                inserted = cur.right;
-              } else {
-                //Traverse the left of the current node
-                if (cur.right) cur = cur.right;
-              }
-            }
-          } else {
-            // TODO may need to support undefined inserted
-          }
-        } else {
-          traversing = false;
-        }
-      }
+    const orgNodeCount = newNode?.count || 0;
+    const inserted = super.add(newNode);
+    if (inserted) {
+      this._count += orgNodeCount;
     }
-    if (inserted) this._balancePath(inserted);
     return inserted;
   }
 
   /**
-   * Time Complexity: O(1) - constant time, as it performs basic pointer assignments.
-   * Space Complexity: O(1) - constant space, as it only uses a constant amount of memory.
+   * Time Complexity: O(k log n) - logarithmic time, where "n" is the number of nodes in the tree. The add method of the superclass (AVLTree) has logarithmic time complexity.
+   * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
    */
 
   /**
-   * Time Complexity: O(k log n) - logarithmic time for each insertion, where "n" is the number of nodes in the tree, and "k" is the number of keys to be inserted. This is because the method iterates through the keys and calls the add method for each.
+   * Time Complexity: O(k log n) - logarithmic time, where "n" is the number of nodes in the tree. The add method of the superclass (AVLTree) has logarithmic time complexity.
    * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
    *
-   * The function `addMany` takes an array of keys or nodes and adds them to the TreeMultimap,
-   * returning an array of the inserted nodes.
-   * @param {(BTNKey | N | undefined)[]} keysOrNodes - An array of keys or nodes. Each element can be
-   * of type BTNKey, N, or undefined.
-   * @param {V[]} [data] - The `data` parameter is an optional array of values that correspond to the
-   * keys or nodes being added. It is used to associate data with each key or node being added to the
-   * TreeMultimap. If provided, the length of the `data` array should be the same as the length of the
-   * @returns The function `addMany` returns an array of nodes (`N`) or `undefined` values.
+   * The function overrides the addMany method to add multiple keys, nodes, or entries to a data
+   * structure.
+   * @param keysOrNodesOrEntries - The parameter `keysOrNodesOrEntries` is an iterable that can contain
+   * either keys, nodes, or entries.
+   * @returns The method is returning an array of type `N | undefined`.
    */
-  override addMany(keysOrNodes: (BTNKey | N | undefined)[], data?: V[]): (N | undefined)[] {
-    const inserted: (N | undefined)[] = [];
-
-    for (let i = 0; i < keysOrNodes.length; i++) {
-      const keyOrNode = keysOrNodes[i];
-
-      if (keyOrNode instanceof TreeMultimapNode) {
-        inserted.push(this.add(keyOrNode.key, keyOrNode.value, keyOrNode.count));
-        continue;
-      }
-
-      if (keyOrNode === undefined) {
-        inserted.push(this.add(NaN, undefined, 0));
-        continue;
-      }
-
-      inserted.push(this.add(keyOrNode, data?.[i], 1));
-    }
-    return inserted;
+  override addMany(keysOrNodesOrEntries: Iterable<BTNodeExemplar<V, N>>): (N | undefined)[] {
+    return super.addMany(keysOrNodesOrEntries);
   }
 
   /**
-   * Time Complexity: O(k log n) - logarithmic time for each insertion, where "n" is the number of nodes in the tree, and "k" is the number of keys to be inserted. This is because the method iterates through the keys and calls the add method for each.
-   * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
+   * Time Complexity: O(1) - constant time, as it performs basic pointer assignments.
+   * Space Complexity: O(1) - constant space, as it only uses a constant amount of memory.
    */
 
   /**
@@ -223,7 +158,7 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
    * values:
    * @returns a boolean value.
    */
-  override perfectlyBalance(iterationType = this.options.iterationType): boolean {
+  override perfectlyBalance(iterationType = this.iterationType): boolean {
     const sorted = this.dfs(node => node, 'in'),
       n = sorted.length;
     if (sorted.length < 1) return false;
@@ -235,7 +170,7 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
         if (l > r) return;
         const m = l + Math.floor((r - l) / 2);
         const midNode = sorted[m];
-        this.add(midNode.key, midNode.value, midNode.count);
+        this.add([midNode.key, midNode.value], midNode.count);
         buildBalanceBST(l, m - 1);
         buildBalanceBST(m + 1, r);
       };
@@ -251,7 +186,7 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
           if (l <= r) {
             const m = l + Math.floor((r - l) / 2);
             const midNode = sorted[m];
-            this.add(midNode.key, midNode.value, midNode.count);
+            this.add([midNode.key, midNode.value], midNode.count);
             stack.push([m + 1, r]);
             stack.push([l, m - 1]);
           }
@@ -262,8 +197,8 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
   }
 
   /**
-   * Time Complexity: O(n log n) - logarithmic time for each insertion, where "n" is the number of nodes in the tree. This is because the method calls the add method for each node.
-   * Space Complexity: O(n) - linear space, as it creates an array to store the sorted nodes.
+   * Time Complexity: O(k log n) - logarithmic time for each insertion, where "n" is the number of nodes in the tree, and "k" is the number of keys to be inserted. This is because the method iterates through the keys and calls the add method for each.
+   * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
    */
 
   /**
@@ -320,7 +255,7 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
         const leftSubTreeRightMost = curr.left ? this.getRightMost(curr.left) : undefined;
         if (leftSubTreeRightMost) {
           const parentOfLeftSubTreeMax = leftSubTreeRightMost.parent;
-          orgCurrent = this._swap(curr, leftSubTreeRightMost);
+          orgCurrent = this._swapProperties(curr, leftSubTreeRightMost);
           if (parentOfLeftSubTreeMax) {
             if (parentOfLeftSubTreeMax.right === leftSubTreeRightMost) {
               parentOfLeftSubTreeMax.right = leftSubTreeRightMost.left;
@@ -346,8 +281,8 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
   }
 
   /**
-   * Time Complexity: O(log n) - logarithmic time, where "n" is the number of nodes in the tree. The delete method of the superclass (AVLTree) has logarithmic time complexity.
-   * Space Complexity: O(1) - constant space, as it doesn't use additional data structures that scale with input size.
+   * Time Complexity: O(n log n) - logarithmic time for each insertion, where "n" is the number of nodes in the tree. This is because the method calls the add method for each node.
+   * Space Complexity: O(n) - linear space, as it creates an array to store the sorted nodes.
    */
 
   /**
@@ -373,8 +308,8 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
    * @returns The method `_addTo` returns either the `parent.left` or `parent.right` node that was
    * added, or `undefined` if no node was added.
    */
-  protected override _addTo(newNode: N | undefined, parent: BTNKey | N | undefined): N | undefined {
-    parent = this.ensureNotKey(parent);
+  protected override _addTo(newNode: N | undefined, parent: BSTNodeKeyOrNode<N>): N | undefined {
+    parent = this.ensureNode(parent);
     if (parent) {
       if (parent.left === undefined) {
         parent.left = newNode;
@@ -400,7 +335,7 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
   }
 
   /**
-   * The `_swap` function swaps the key, value, count, and height properties between two nodes.
+   * The `_swapProperties` function swaps the key, value, count, and height properties between two nodes.
    * @param {BTNKey | N | undefined} srcNode - The `srcNode` parameter represents the source node from
    * which the values will be swapped. It can be of type `BTNKey`, `N`, or `undefined`.
    * @param {BTNKey | N | undefined} destNode - The `destNode` parameter represents the destination
@@ -408,9 +343,9 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
    * @returns either the `destNode` object if both `srcNode` and `destNode` are defined, or `undefined`
    * if either `srcNode` or `destNode` is undefined.
    */
-  protected _swap(srcNode: BTNKey | N | undefined, destNode: BTNKey | N | undefined): N | undefined {
-    srcNode = this.ensureNotKey(srcNode);
-    destNode = this.ensureNotKey(destNode);
+  protected override _swapProperties(srcNode: BSTNodeKeyOrNode<N>, destNode: BSTNodeKeyOrNode<N>): N | undefined {
+    srcNode = this.ensureNode(srcNode);
+    destNode = this.ensureNode(destNode);
     if (srcNode && destNode) {
       const { key, value, count, height } = destNode;
       const tempNode = this.createNode(key, value, count);
@@ -432,4 +367,9 @@ export class TreeMultimap<V = any, N extends TreeMultimapNode<V, N> = TreeMultim
     }
     return undefined;
   }
+
+  protected _replaceNode(oldNode: N, newNode: N): N {
+    newNode.count = oldNode.count + newNode.count
+    return super._replaceNode(oldNode, newNode);
+  }
 }

package/src/data-structures/graph/abstract-graph.ts

@@ -691,7 +691,7 @@ export abstract class AbstractGraph<
       if (vertexOrKey instanceof AbstractVertex) distMap.set(vertexOrKey, Infinity);
     }
 
-    const heap = new PriorityQueue<{ key: number; value: VO }>({ comparator: (a, b) => a.key - b.key });
+    const heap = new PriorityQueue<{ key: number; value: VO }>([], { comparator: (a, b) => a.key - b.key });
     heap.add({ key: 0, value: srcVertex });
 
     distMap.set(srcVertex, 0);

package/src/data-structures/hash/hash-map.ts

@@ -19,20 +19,16 @@ export class HashMap<K = any, V = any> {
   protected _hashFn: (key: K) => string;
   protected _objHashFn: (key: K) => object;
 
-  /**
-   * The constructor initializes a HashMapLinkedNode with an optional iterable of key-value pairs.
-   * @param options - The `options` parameter is an object that contains the `elements` property. The
-   * `elements` property is an iterable that contains key-value pairs represented as arrays `[K, V]`.
-   */
-  constructor(options: HashMapOptions<K, V> = {
-    elements: [],
+
+  constructor(elements?: Iterable<[K, V]>, options: HashMapOptions<K> = {
+
     hashFn: (key: K) => String(key),
     objHashFn: (key: K) => (<object>key)
   }) {
     this._sentinel = <HashMapLinkedNode<K, V>>{};
     this._sentinel.prev = this._sentinel.next = this._head = this._tail = this._sentinel;
 
-    const { elements, hashFn, objHashFn } = options;
+    const { hashFn, objHashFn } = options;
     this._hashFn = hashFn;
     this._objHashFn = objHashFn;
     if (elements) {
@@ -379,6 +375,10 @@ export class HashMap<K = any, V = any> {
     }
   }
 
+  print() {
+    console.log([...this]);
+  }
+
   /**
    * Time Complexity: O(1)
    * Space Complexity: O(1)