red-black-tree-typed 2.1.2 → 2.2.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (37) hide show
  1. package/dist/cjs/index.cjs +329 -177
  2. package/dist/cjs/index.cjs.map +1 -1
  3. package/dist/cjs-legacy/index.cjs +4019 -0
  4. package/dist/cjs-legacy/index.cjs.map +1 -0
  5. package/dist/esm/index.mjs +329 -177
  6. package/dist/esm/index.mjs.map +1 -1
  7. package/dist/esm-legacy/index.mjs +4010 -0
  8. package/dist/esm-legacy/index.mjs.map +1 -0
  9. package/dist/types/data-structures/binary-tree/avl-tree-counter.d.ts +57 -3
  10. package/dist/types/data-structures/binary-tree/avl-tree-multi-map.d.ts +65 -3
  11. package/dist/types/data-structures/binary-tree/avl-tree.d.ts +61 -5
  12. package/dist/types/data-structures/binary-tree/binary-tree.d.ts +1 -0
  13. package/dist/types/data-structures/binary-tree/bst.d.ts +58 -3
  14. package/dist/types/data-structures/binary-tree/red-black-tree.d.ts +59 -4
  15. package/dist/types/data-structures/binary-tree/tree-counter.d.ts +57 -3
  16. package/dist/types/data-structures/binary-tree/tree-multi-map.d.ts +66 -3
  17. package/dist/types/types/data-structures/base/base.d.ts +1 -1
  18. package/dist/umd/red-black-tree-typed.js +176 -22
  19. package/dist/umd/red-black-tree-typed.js.map +1 -1
  20. package/dist/umd/red-black-tree-typed.min.js +3 -3
  21. package/dist/umd/red-black-tree-typed.min.js.map +1 -1
  22. package/package.json +20 -2
  23. package/src/data-structures/base/iterable-entry-base.ts +4 -4
  24. package/src/data-structures/binary-tree/avl-tree-counter.ts +103 -12
  25. package/src/data-structures/binary-tree/avl-tree-multi-map.ts +116 -12
  26. package/src/data-structures/binary-tree/avl-tree.ts +109 -16
  27. package/src/data-structures/binary-tree/binary-tree.ts +3 -2
  28. package/src/data-structures/binary-tree/bst.ts +104 -12
  29. package/src/data-structures/binary-tree/red-black-tree.ts +110 -19
  30. package/src/data-structures/binary-tree/tree-counter.ts +102 -11
  31. package/src/data-structures/binary-tree/tree-multi-map.ts +124 -12
  32. package/src/data-structures/graph/abstract-graph.ts +8 -8
  33. package/src/data-structures/graph/directed-graph.ts +5 -5
  34. package/src/data-structures/graph/undirected-graph.ts +5 -5
  35. package/src/data-structures/hash/hash-map.ts +4 -4
  36. package/src/types/data-structures/base/base.ts +1 -1
  37. package/tsup.node.config.js +40 -6
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "red-black-tree-typed",
3
- "version": "2.1.2",
3
+ "version": "2.2.1",
4
4
  "description": "red black tree",
5
5
  "browser": "dist/umd/red-black-tree-typed.min.js",
6
6
  "umd:main": "dist/umd/red-black-tree-typed.min.js",
@@ -9,9 +9,27 @@
9
9
  "types": "dist/types/index.d.ts",
10
10
  "exports": {
11
11
  ".": {
12
+ "types": "./dist/types/index.d.ts",
13
+ "node": {
14
+ "import": "./dist/esm/index.mjs",
15
+ "require": "./dist/cjs/index.cjs"
16
+ },
17
+ "browser": {
18
+ "import": "./dist/esm-legacy/index.mjs",
19
+ "require": "./dist/cjs-legacy/index.cjs"
20
+ },
21
+ "import": "./dist/esm-legacy/index.mjs",
22
+ "require": "./dist/cjs-legacy/index.cjs"
23
+ },
24
+ "./modern": {
12
25
  "types": "./dist/types/index.d.ts",
13
26
  "import": "./dist/esm/index.mjs",
14
27
  "require": "./dist/cjs/index.cjs"
28
+ },
29
+ "./legacy": {
30
+ "types": "./dist/types/index.d.ts",
31
+ "import": "./dist/esm-legacy/index.mjs",
32
+ "require": "./dist/cjs-legacy/index.cjs"
15
33
  }
16
34
  },
17
35
  "sideEffects": false,
@@ -162,6 +180,6 @@
162
180
  "typescript": "^4.9.5"
163
181
  },
164
182
  "dependencies": {
165
- "data-structure-typed": "^2.1.2"
183
+ "data-structure-typed": "^2.2.1"
166
184
  }
167
185
  }
@@ -66,7 +66,7 @@ export abstract class IterableEntryBase<K = any, V = any> {
66
66
  every(predicate: EntryCallback<K, V, boolean>, thisArg?: any): boolean {
67
67
  let index = 0;
68
68
  for (const item of this) {
69
- if (!predicate.call(thisArg, item[0], item[1], index++, this)) {
69
+ if (!predicate.call(thisArg, item[1], item[0], index++, this)) {
70
70
  return false;
71
71
  }
72
72
  }
@@ -83,7 +83,7 @@ export abstract class IterableEntryBase<K = any, V = any> {
83
83
  some(predicate: EntryCallback<K, V, boolean>, thisArg?: any): boolean {
84
84
  let index = 0;
85
85
  for (const item of this) {
86
- if (predicate.call(thisArg, item[0], item[1], index++, this)) {
86
+ if (predicate.call(thisArg, item[1], item[0], index++, this)) {
87
87
  return true;
88
88
  }
89
89
  }
@@ -100,7 +100,7 @@ export abstract class IterableEntryBase<K = any, V = any> {
100
100
  let index = 0;
101
101
  for (const item of this) {
102
102
  const [key, value] = item;
103
- callbackfn.call(thisArg, key, value, index++, this);
103
+ callbackfn.call(thisArg, value, key, index++, this);
104
104
  }
105
105
  }
106
106
 
@@ -115,7 +115,7 @@ export abstract class IterableEntryBase<K = any, V = any> {
115
115
  let index = 0;
116
116
  for (const item of this) {
117
117
  const [key, value] = item;
118
- if (callbackfn.call(thisArg, key, value, index++, this)) return item;
118
+ if (callbackfn.call(thisArg, value, key, index++, this)) return item;
119
119
  }
120
120
  return;
121
121
  }
@@ -12,10 +12,12 @@ import type {
12
12
  BinaryTreeOptions,
13
13
  BSTNOptKeyOrNode,
14
14
  EntryCallback,
15
- IterationType
15
+ FamilyPosition,
16
+ IterationType,
17
+ RBTNColor
16
18
  } from '../../types';
17
19
  import { IBinaryTree } from '../../interfaces';
18
- import { AVLTree, AVLTreeNode } from './avl-tree';
20
+ import { AVLTree } from './avl-tree';
19
21
 
20
22
  /**
21
23
  * AVL node with an extra 'count' field; keeps parent/child links.
@@ -23,8 +25,10 @@ import { AVLTree, AVLTreeNode } from './avl-tree';
23
25
  * @template K
24
26
  * @template V
25
27
  */
26
- export class AVLTreeCounterNode<K = any, V = any> extends AVLTreeNode<K, V> {
27
- override parent?: AVLTreeCounterNode<K, V> = undefined;
28
+ export class AVLTreeCounterNode<K = any, V = any> {
29
+ key: K;
30
+ value?: V;
31
+ parent?: AVLTreeCounterNode<K, V> = undefined;
28
32
 
29
33
  /**
30
34
  * Create an AVL counter node.
@@ -35,18 +39,19 @@ export class AVLTreeCounterNode<K = any, V = any> extends AVLTreeNode<K, V> {
35
39
  * @returns New AVLTreeCounterNode instance.
36
40
  */
37
41
  constructor(key: K, value?: V, count = 1) {
38
- super(key, value);
42
+ this.key = key;
43
+ this.value = value;
39
44
  this.count = count;
40
45
  }
41
46
 
42
- override _left?: AVLTreeCounterNode<K, V> | null | undefined = undefined;
47
+ _left?: AVLTreeCounterNode<K, V> | null | undefined = undefined;
43
48
 
44
49
  /**
45
50
  * Get the left child pointer.
46
51
  * @remarks Time O(1), Space O(1)
47
52
  * @returns Left child node, or null/undefined.
48
53
  */
49
- override get left(): AVLTreeCounterNode<K, V> | null | undefined {
54
+ get left(): AVLTreeCounterNode<K, V> | null | undefined {
50
55
  return this._left;
51
56
  }
52
57
 
@@ -56,21 +61,21 @@ export class AVLTreeCounterNode<K = any, V = any> extends AVLTreeNode<K, V> {
56
61
  * @param v - New left child node, or null/undefined.
57
62
  * @returns void
58
63
  */
59
- override set left(v: AVLTreeCounterNode<K, V> | null | undefined) {
64
+ set left(v: AVLTreeCounterNode<K, V> | null | undefined) {
60
65
  if (v) {
61
66
  v.parent = this;
62
67
  }
63
68
  this._left = v;
64
69
  }
65
70
 
66
- override _right?: AVLTreeCounterNode<K, V> | null | undefined = undefined;
71
+ _right?: AVLTreeCounterNode<K, V> | null | undefined = undefined;
67
72
 
68
73
  /**
69
74
  * Get the right child pointer.
70
75
  * @remarks Time O(1), Space O(1)
71
76
  * @returns Right child node, or null/undefined.
72
77
  */
73
- override get right(): AVLTreeCounterNode<K, V> | null | undefined {
78
+ get right(): AVLTreeCounterNode<K, V> | null | undefined {
74
79
  return this._right;
75
80
  }
76
81
 
@@ -80,12 +85,98 @@ export class AVLTreeCounterNode<K = any, V = any> extends AVLTreeNode<K, V> {
80
85
  * @param v - New right child node, or null/undefined.
81
86
  * @returns void
82
87
  */
83
- override set right(v: AVLTreeCounterNode<K, V> | null | undefined) {
88
+ set right(v: AVLTreeCounterNode<K, V> | null | undefined) {
84
89
  if (v) {
85
90
  v.parent = this;
86
91
  }
87
92
  this._right = v;
88
93
  }
94
+
95
+ _height: number = 0;
96
+
97
+ /**
98
+ * Gets the height of the node (used in self-balancing trees).
99
+ * @remarks Time O(1), Space O(1)
100
+ *
101
+ * @returns The height.
102
+ */
103
+ get height(): number {
104
+ return this._height;
105
+ }
106
+
107
+ /**
108
+ * Sets the height of the node.
109
+ * @remarks Time O(1), Space O(1)
110
+ *
111
+ * @param value - The new height.
112
+ */
113
+ set height(value: number) {
114
+ this._height = value;
115
+ }
116
+
117
+ _color: RBTNColor = 'BLACK';
118
+
119
+ /**
120
+ * Gets the color of the node (used in Red-Black trees).
121
+ * @remarks Time O(1), Space O(1)
122
+ *
123
+ * @returns The node's color.
124
+ */
125
+ get color(): RBTNColor {
126
+ return this._color;
127
+ }
128
+
129
+ /**
130
+ * Sets the color of the node.
131
+ * @remarks Time O(1), Space O(1)
132
+ *
133
+ * @param value - The new color.
134
+ */
135
+ set color(value: RBTNColor) {
136
+ this._color = value;
137
+ }
138
+
139
+ _count: number = 1;
140
+
141
+ /**
142
+ * Gets the count of nodes in the subtree rooted at this node (used in order-statistic trees).
143
+ * @remarks Time O(1), Space O(1)
144
+ *
145
+ * @returns The subtree node count.
146
+ */
147
+ get count(): number {
148
+ return this._count;
149
+ }
150
+
151
+ /**
152
+ * Sets the count of nodes in the subtree.
153
+ * @remarks Time O(1), Space O(1)
154
+ *
155
+ * @param value - The new count.
156
+ */
157
+ set count(value: number) {
158
+ this._count = value;
159
+ }
160
+
161
+ /**
162
+ * Gets the position of the node relative to its parent.
163
+ * @remarks Time O(1), Space O(1)
164
+ *
165
+ * @returns The family position (e.g., 'ROOT', 'LEFT', 'RIGHT').
166
+ */
167
+ get familyPosition(): FamilyPosition {
168
+ if (!this.parent) {
169
+ return this.left || this.right ? 'ROOT' : 'ISOLATED';
170
+ }
171
+
172
+ if (this.parent.left === this) {
173
+ return this.left || this.right ? 'ROOT_LEFT' : 'LEFT';
174
+ } else if (this.parent.right === this) {
175
+ return this.left || this.right ? 'ROOT_RIGHT' : 'RIGHT';
176
+ }
177
+
178
+ return 'MAL_NODE';
179
+ }
89
180
  }
90
181
 
91
182
  /**
@@ -320,7 +411,7 @@ export class AVLTreeCounter<K = any, V = any, R = any> extends AVLTree<K, V, R>
320
411
 
321
412
  let index = 0;
322
413
  for (const [key, value] of this) {
323
- out.add(callback.call(thisArg, key, value, index++, this));
414
+ out.add(callback.call(thisArg, value, key, index++, this));
324
415
  }
325
416
  return out;
326
417
  }
@@ -12,7 +12,9 @@ import type {
12
12
  BTNOptKeyOrNull,
13
13
  ElemOf,
14
14
  EntryCallback,
15
- IterationType
15
+ FamilyPosition,
16
+ IterationType,
17
+ RBTNColor
16
18
  } from '../../types';
17
19
  import { AVLTree, AVLTreeNode } from './avl-tree';
18
20
  import { IBinaryTree } from '../../interfaces';
@@ -23,8 +25,10 @@ import { IBinaryTree } from '../../interfaces';
23
25
  * @template K
24
26
  * @template V
25
27
  */
26
- export class AVLTreeMultiMapNode<K = any, V = any> extends AVLTreeNode<K, V[]> {
27
- override parent?: AVLTreeMultiMapNode<K, V> = undefined;
28
+ export class AVLTreeMultiMapNode<K = any, V = any> {
29
+ key: K;
30
+ value?: V[];
31
+ parent?: AVLTreeMultiMapNode<K, V> = undefined;
28
32
 
29
33
  /**
30
34
  * Create an AVLTreeMultiMap node with a value bucket.
@@ -33,18 +37,19 @@ export class AVLTreeMultiMapNode<K = any, V = any> extends AVLTreeNode<K, V[]> {
33
37
  * @param value - Initial array of values.
34
38
  * @returns New AVLTreeMultiMapNode instance.
35
39
  */
36
- constructor(key: K, value: V[]) {
37
- super(key, value);
40
+ constructor(key: K, value: V[] = []) {
41
+ this.key = key;
42
+ this.value = value;
38
43
  }
39
44
 
40
- override _left?: AVLTreeMultiMapNode<K, V> | null | undefined = undefined;
45
+ _left?: AVLTreeMultiMapNode<K, V> | null | undefined = undefined;
41
46
 
42
47
  /**
43
48
  * Get the left child pointer.
44
49
  * @remarks Time O(1), Space O(1)
45
50
  * @returns Left child node, or null/undefined.
46
51
  */
47
- override get left(): AVLTreeMultiMapNode<K, V> | null | undefined {
52
+ get left(): AVLTreeMultiMapNode<K, V> | null | undefined {
48
53
  return this._left;
49
54
  }
50
55
 
@@ -54,21 +59,21 @@ export class AVLTreeMultiMapNode<K = any, V = any> extends AVLTreeNode<K, V[]> {
54
59
  * @param v - New left child node, or null/undefined.
55
60
  * @returns void
56
61
  */
57
- override set left(v: AVLTreeMultiMapNode<K, V> | null | undefined) {
62
+ set left(v: AVLTreeMultiMapNode<K, V> | null | undefined) {
58
63
  if (v) {
59
64
  v.parent = this;
60
65
  }
61
66
  this._left = v;
62
67
  }
63
68
 
64
- override _right?: AVLTreeMultiMapNode<K, V> | null | undefined = undefined;
69
+ _right?: AVLTreeMultiMapNode<K, V> | null | undefined = undefined;
65
70
 
66
71
  /**
67
72
  * Get the right child pointer.
68
73
  * @remarks Time O(1), Space O(1)
69
74
  * @returns Right child node, or null/undefined.
70
75
  */
71
- override get right(): AVLTreeMultiMapNode<K, V> | null | undefined {
76
+ get right(): AVLTreeMultiMapNode<K, V> | null | undefined {
72
77
  return this._right;
73
78
  }
74
79
 
@@ -78,12 +83,98 @@ export class AVLTreeMultiMapNode<K = any, V = any> extends AVLTreeNode<K, V[]> {
78
83
  * @param v - New right child node, or null/undefined.
79
84
  * @returns void
80
85
  */
81
- override set right(v: AVLTreeMultiMapNode<K, V> | null | undefined) {
86
+ set right(v: AVLTreeMultiMapNode<K, V> | null | undefined) {
82
87
  if (v) {
83
88
  v.parent = this;
84
89
  }
85
90
  this._right = v;
86
91
  }
92
+
93
+ _height: number = 0;
94
+
95
+ /**
96
+ * Gets the height of the node (used in self-balancing trees).
97
+ * @remarks Time O(1), Space O(1)
98
+ *
99
+ * @returns The height.
100
+ */
101
+ get height(): number {
102
+ return this._height;
103
+ }
104
+
105
+ /**
106
+ * Sets the height of the node.
107
+ * @remarks Time O(1), Space O(1)
108
+ *
109
+ * @param value - The new height.
110
+ */
111
+ set height(value: number) {
112
+ this._height = value;
113
+ }
114
+
115
+ _color: RBTNColor = 'BLACK';
116
+
117
+ /**
118
+ * Gets the color of the node (used in Red-Black trees).
119
+ * @remarks Time O(1), Space O(1)
120
+ *
121
+ * @returns The node's color.
122
+ */
123
+ get color(): RBTNColor {
124
+ return this._color;
125
+ }
126
+
127
+ /**
128
+ * Sets the color of the node.
129
+ * @remarks Time O(1), Space O(1)
130
+ *
131
+ * @param value - The new color.
132
+ */
133
+ set color(value: RBTNColor) {
134
+ this._color = value;
135
+ }
136
+
137
+ _count: number = 1;
138
+
139
+ /**
140
+ * Gets the count of nodes in the subtree rooted at this node (used in order-statistic trees).
141
+ * @remarks Time O(1), Space O(1)
142
+ *
143
+ * @returns The subtree node count.
144
+ */
145
+ get count(): number {
146
+ return this._count;
147
+ }
148
+
149
+ /**
150
+ * Sets the count of nodes in the subtree.
151
+ * @remarks Time O(1), Space O(1)
152
+ *
153
+ * @param value - The new count.
154
+ */
155
+ set count(value: number) {
156
+ this._count = value;
157
+ }
158
+
159
+ /**
160
+ * Gets the position of the node relative to its parent.
161
+ * @remarks Time O(1), Space O(1)
162
+ *
163
+ * @returns The family position (e.g., 'ROOT', 'LEFT', 'RIGHT').
164
+ */
165
+ get familyPosition(): FamilyPosition {
166
+ if (!this.parent) {
167
+ return this.left || this.right ? 'ROOT' : 'ISOLATED';
168
+ }
169
+
170
+ if (this.parent.left === this) {
171
+ return this.left || this.right ? 'ROOT_LEFT' : 'LEFT';
172
+ } else if (this.parent.right === this) {
173
+ return this.left || this.right ? 'ROOT_RIGHT' : 'RIGHT';
174
+ }
175
+
176
+ return 'MAL_NODE';
177
+ }
87
178
  }
88
179
 
89
180
  /**
@@ -117,6 +208,19 @@ export class AVLTreeMultiMap<K = any, V = any, R = any> extends AVLTree<K, V[],
117
208
  return new AVLTreeMultiMapNode<K, V>(key, this._isMapMode ? [] : value);
118
209
  }
119
210
 
211
+ /**
212
+ * Checks if the given item is a `AVLTreeMultiMapNode` instance.
213
+ * @remarks Time O(1), Space O(1)
214
+ *
215
+ * @param keyNodeOrEntry - The item to check.
216
+ * @returns True if it's a AVLTreeMultiMapNode, false otherwise.
217
+ */
218
+ override isNode(
219
+ keyNodeOrEntry: K | AVLTreeMultiMapNode<K, V> | [K | null | undefined, V[] | undefined] | null | undefined
220
+ ): keyNodeOrEntry is AVLTreeMultiMapNode<K, V> {
221
+ return keyNodeOrEntry instanceof AVLTreeMultiMapNode;
222
+ }
223
+
120
224
  override add(
121
225
  keyNodeOrEntry: K | AVLTreeMultiMapNode<K, V> | [K | null | undefined, V[] | undefined] | null | undefined
122
226
  ): boolean;
@@ -289,7 +393,7 @@ export class AVLTreeMultiMap<K = any, V = any, R = any> extends AVLTree<K, V[],
289
393
  ): AVLTree<MK, MV, MR> {
290
394
  const out = this._createLike<MK, MV, MR>([], options);
291
395
  let i = 0;
292
- for (const [k, v] of this) out.add(callback.call(thisArg, k, v, i++, this));
396
+ for (const [k, v] of this) out.add(callback.call(thisArg, v, k, i++, this));
293
397
  return out;
294
398
  }
295
399
 
@@ -6,14 +6,16 @@
6
6
  * @license MIT License
7
7
  */
8
8
 
9
- import { BST, BSTNode } from './bst';
9
+ import { BST } from './bst';
10
10
  import type {
11
11
  AVLTreeOptions,
12
12
  BinaryTreeDeleteResult,
13
13
  BinaryTreeOptions,
14
14
  BSTNOptKeyOrNode,
15
15
  EntryCallback,
16
- IterationType
16
+ FamilyPosition,
17
+ IterationType,
18
+ RBTNColor
17
19
  } from '../../types';
18
20
  import { BSTOptions } from '../../types';
19
21
  import { IBinaryTree } from '../../interfaces';
@@ -25,8 +27,10 @@ import { IBinaryTree } from '../../interfaces';
25
27
  * @template K - The type of the key.
26
28
  * @template V - The type of the value.
27
29
  */
28
- export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
29
- override parent?: AVLTreeNode<K, V> = undefined;
30
+ export class AVLTreeNode<K = any, V = any> {
31
+ key: K;
32
+ value?: V;
33
+ parent?: AVLTreeNode<K, V> = undefined;
30
34
 
31
35
  /**
32
36
  * Creates an instance of AVLTreeNode.
@@ -36,10 +40,11 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
36
40
  * @param [value] - The value associated with the key.
37
41
  */
38
42
  constructor(key: K, value?: V) {
39
- super(key, value);
43
+ this.key = key;
44
+ this.value = value;
40
45
  }
41
46
 
42
- override _left?: AVLTreeNode<K, V> | null | undefined = undefined;
47
+ _left?: AVLTreeNode<K, V> | null | undefined = undefined;
43
48
 
44
49
  /**
45
50
  * Gets the left child of the node.
@@ -47,7 +52,7 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
47
52
  *
48
53
  * @returns The left child.
49
54
  */
50
- override get left(): AVLTreeNode<K, V> | null | undefined {
55
+ get left(): AVLTreeNode<K, V> | null | undefined {
51
56
  return this._left;
52
57
  }
53
58
 
@@ -57,14 +62,14 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
57
62
  *
58
63
  * @param v - The node to set as the left child.
59
64
  */
60
- override set left(v: AVLTreeNode<K, V> | null | undefined) {
65
+ set left(v: AVLTreeNode<K, V> | null | undefined) {
61
66
  if (v) {
62
67
  v.parent = this;
63
68
  }
64
69
  this._left = v;
65
70
  }
66
71
 
67
- override _right?: AVLTreeNode<K, V> | null | undefined = undefined;
72
+ _right?: AVLTreeNode<K, V> | null | undefined = undefined;
68
73
 
69
74
  /**
70
75
  * Gets the right child of the node.
@@ -72,7 +77,7 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
72
77
  *
73
78
  * @returns The right child.
74
79
  */
75
- override get right(): AVLTreeNode<K, V> | null | undefined {
80
+ get right(): AVLTreeNode<K, V> | null | undefined {
76
81
  return this._right;
77
82
  }
78
83
 
@@ -82,12 +87,98 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
82
87
  *
83
88
  * @param v - The node to set as the right child.
84
89
  */
85
- override set right(v: AVLTreeNode<K, V> | null | undefined) {
90
+ set right(v: AVLTreeNode<K, V> | null | undefined) {
86
91
  if (v) {
87
92
  v.parent = this;
88
93
  }
89
94
  this._right = v;
90
95
  }
96
+
97
+ _height: number = 0;
98
+
99
+ /**
100
+ * Gets the height of the node (used in self-balancing trees).
101
+ * @remarks Time O(1), Space O(1)
102
+ *
103
+ * @returns The height.
104
+ */
105
+ get height(): number {
106
+ return this._height;
107
+ }
108
+
109
+ /**
110
+ * Sets the height of the node.
111
+ * @remarks Time O(1), Space O(1)
112
+ *
113
+ * @param value - The new height.
114
+ */
115
+ set height(value: number) {
116
+ this._height = value;
117
+ }
118
+
119
+ _color: RBTNColor = 'BLACK';
120
+
121
+ /**
122
+ * Gets the color of the node (used in Red-Black trees).
123
+ * @remarks Time O(1), Space O(1)
124
+ *
125
+ * @returns The node's color.
126
+ */
127
+ get color(): RBTNColor {
128
+ return this._color;
129
+ }
130
+
131
+ /**
132
+ * Sets the color of the node.
133
+ * @remarks Time O(1), Space O(1)
134
+ *
135
+ * @param value - The new color.
136
+ */
137
+ set color(value: RBTNColor) {
138
+ this._color = value;
139
+ }
140
+
141
+ _count: number = 1;
142
+
143
+ /**
144
+ * Gets the count of nodes in the subtree rooted at this node (used in order-statistic trees).
145
+ * @remarks Time O(1), Space O(1)
146
+ *
147
+ * @returns The subtree node count.
148
+ */
149
+ get count(): number {
150
+ return this._count;
151
+ }
152
+
153
+ /**
154
+ * Sets the count of nodes in the subtree.
155
+ * @remarks Time O(1), Space O(1)
156
+ *
157
+ * @param value - The new count.
158
+ */
159
+ set count(value: number) {
160
+ this._count = value;
161
+ }
162
+
163
+ /**
164
+ * Gets the position of the node relative to its parent.
165
+ * @remarks Time O(1), Space O(1)
166
+ *
167
+ * @returns The family position (e.g., 'ROOT', 'LEFT', 'RIGHT').
168
+ */
169
+ get familyPosition(): FamilyPosition {
170
+ if (!this.parent) {
171
+ return this.left || this.right ? 'ROOT' : 'ISOLATED';
172
+ }
173
+
174
+ if (this.parent.left === this) {
175
+ return this.left || this.right ? 'ROOT_LEFT' : 'LEFT';
176
+ } else if (this.parent.right === this) {
177
+ return this.left || this.right ? 'ROOT_RIGHT' : 'RIGHT';
178
+ }
179
+
180
+ return 'MAL_NODE';
181
+ }
91
182
  }
92
183
 
93
184
  /**
@@ -104,7 +195,9 @@ export class AVLTreeNode<K = any, V = any> extends BSTNode<K, V> {
104
195
  * 4. Order Preservation: Maintains the binary search tree property where left child values are less than the parent, and right child values are greater.
105
196
  * 5. Efficient Lookups: Offers O(log n) search time, where 'n' is the number of nodes, due to its balanced nature.
106
197
  * 6. Complex Insertions and Deletions: Due to rebalancing, these operations are more complex than in a regular BST.
107
- * 7. Path Length: The path length from the root to any leaf is longer compared to an unbalanced BST, but shorter than a linear chain of nodes.@example
198
+ * 7. Path Length: The path length from the root to any leaf is longer compared to an unbalanced BST, but shorter than a linear chain of nodes.
199
+ *
200
+ * @example
108
201
  * // Find elements in a range
109
202
  * // In interval queries, AVL trees, with their strictly balanced structure and lower height, offer better query efficiency, making them ideal for frequent and high-performance interval queries. In contrast, Red-Black trees, with lower update costs, are more suitable for scenarios involving frequent insertions and deletions where the requirements for interval queries are less demanding.
110
203
  * type Datum = { timestamp: Date; temperature: number };
@@ -312,7 +405,7 @@ export class AVLTree<K = any, V = any, R = any> extends BST<K, V, R> implements
312
405
  // Iterates in-order
313
406
  for (const [key, value] of this) {
314
407
  // `add` on the new tree will be O(log N) and will self-balance.
315
- out.add(callback.call(thisArg, key, value, index++, this));
408
+ out.add(callback.call(thisArg, value, key, index++, this));
316
409
  }
317
410
  return out;
318
411
  }
@@ -327,7 +420,7 @@ export class AVLTree<K = any, V = any, R = any> extends BST<K, V, R> implements
327
420
  */
328
421
  protected override _createInstance<TK = K, TV = V, TR = R>(options?: Partial<BSTOptions<TK, TV, TR>>): this {
329
422
  const Ctor = this.constructor as unknown as new (
330
- iter?: Iterable<TK | BSTNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
423
+ iter?: Iterable<TK | AVLTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
331
424
  opts?: BSTOptions<TK, TV, TR>
332
425
  ) => this;
333
426
  return new Ctor([], { ...this._snapshotOptions<TK, TV, TR>(), ...(options ?? {}) }) as unknown as this;
@@ -343,11 +436,11 @@ export class AVLTree<K = any, V = any, R = any> extends BST<K, V, R> implements
343
436
  * @returns A new AVLTree.
344
437
  */
345
438
  protected override _createLike<TK = K, TV = V, TR = R>(
346
- iter: Iterable<TK | BSTNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR> = [],
439
+ iter: Iterable<TK | AVLTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR> = [],
347
440
  options?: Partial<BSTOptions<TK, TV, TR>>
348
441
  ): AVLTree<TK, TV, TR> {
349
442
  const Ctor = this.constructor as unknown as new (
350
- iter?: Iterable<TK | BSTNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
443
+ iter?: Iterable<TK | AVLTreeNode<TK, TV> | [TK | null | undefined, TV | undefined] | null | undefined | TR>,
351
444
  opts?: BSTOptions<TK, TV, TR>
352
445
  ) => AVLTree<TK, TV, TR>;
353
446
  return new Ctor(iter, { ...this._snapshotOptions<TK, TV, TR>(), ...(options ?? {}) });