x/data-structure-typed
1
0
Fork 0
mirror of https://github.com/zrwusa/data-structure-typed.git synced 2024-11-23 12:54:04 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[binary-tree] Move the print method inside BinaryTree. [rb-tree] Replace sentinel nodes with NaN. [bst] Replace all direct assignments to _root with calls to the _setRoot method.

Browse source
This commit is contained in:
Revone 2023-11-07 17:41:51 +08:00
parent ea89c0278b
commit b78d92fcd1
4 changed files with 88 additions and 97 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
src/data-structures/binary-tree/binary-tree.ts
View file

@ -1484,5 +1484,63 @@ export class BinaryTree<V = any, N extends BinaryTreeNode<V, N> = BinaryTreeNode
this._root = v;
}
print(beginRoot: N | null | undefined = this.root) {
const display = (root: N | null | undefined): void => {
const [lines, , ,] = _displayAux(root);
for (const line of lines) {
console.log(line);
}
};
const _displayAux = (node: N | null | undefined): [string[], number, number, number] => {
if (node === undefined || node === null) {
return [[], 0, 0, 0];
}
if (node && node.right === undefined && node.left === undefined) {
const line = `${node.key}`;
const width = line.length;
const height = 1;
const middle = Math.floor(width / 2);
return [[line], width, height, middle];
}
if (node && node.right === undefined) {
const [lines, n, p, x] = _displayAux(node.left);
const s = `${node.key}`;
const u = s.length;
const first_line = ' '.repeat(x + 1) + '_'.repeat(n - x - 1) + s;
const second_line = ' '.repeat(x) + '/' + ' '.repeat(n - x - 1 + u);
const shifted_lines = lines.map(line => line + ' '.repeat(u));
return [[first_line, second_line, ...shifted_lines], n + u, p + 2, n + Math.floor(u / 2)];
}
if (node && node.left === undefined) {
const [lines, n, p, u] = _displayAux(node.right);
const s = `${node.key}`;
const x = s.length;
const first_line = s + '_'.repeat(x) + ' '.repeat(n - x);
const second_line = ' '.repeat(u + x) + '\\' + ' '.repeat(n - x - 1);
const shifted_lines = lines.map(line => ' '.repeat(u) + line);
return [[first_line, second_line, ...shifted_lines], n + x, p + 2, Math.floor(u / 2)];
}
const [left, n, p, x] = _displayAux(node.left);
const [right, m, q, y] = _displayAux(node.right);
const s = `${node.key}`;
const u = s.length;
const first_line = ' '.repeat(x + 1) + '_'.repeat(n - x - 1) + s + '_'.repeat(y) + ' '.repeat(m - y);
const second_line = ' '.repeat(x) + '/' + ' '.repeat(n - x - 1 + u + y) + '\\' + ' '.repeat(m - y - 1);
if (p < q) {
left.push(...new Array(q - p).fill(' '.repeat(n)));
} else if (q < p) {
right.push(...new Array(p - q).fill(' '.repeat(m)));
}
const zipped_lines = left.map((a, i) => a + ' '.repeat(u) + right[i]);
return [[first_line, second_line, ...zipped_lines], n + m + u, Math.max(p, q) + 2, n + Math.floor(u / 2)];
};
display(beginRoot);
}
// --- end additional methods ---
}

13
src/data-structures/binary-tree/bst.ts
View file

@ -119,8 +119,8 @@ export class BST<V = any, N extends BSTNode<V, N> = BSTNode<V, BSTNodeNested<V>>
}
if (keyOrNode === null) return undefined;
// TODO support node as a parameter
let inserted:N | undefined = undefined;
let newNode:N | undefined = undefined;
let inserted:N | undefined;
let newNode:N | undefined;
if (keyOrNode instanceof BSTNode) {
newNode = keyOrNode;
} else if (typeof keyOrNode === 'number') {
@ -129,7 +129,7 @@ export class BST<V = any, N extends BSTNode<V, N> = BSTNode<V, BSTNodeNested<V>>
newNode = undefined;
}
if (this.root === undefined) {
this._root = newNode;
this._setRoot(newNode);
this._size = this.size + 1;
inserted = this.root;
} else {
@ -538,6 +538,13 @@ export class BST<V = any, N extends BSTNode<V, N> = BSTNode<V, BSTNodeNested<V>>
protected _comparator: BSTComparator = (a, b) => a - b;
protected _setRoot(v: N | undefined) {
if (v) {
v.parent = undefined;
}
this._root = v;
}
/**
* The function compares two values using a comparator function and returns whether the first value
* is greater than, less than, or equal to the second value.

100
src/data-structures/binary-tree/rb-tree.ts
View file

@ -56,7 +56,7 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
return this._size;
}
NIL: N = new RBTreeNode<V>(0) as unknown as N;
NIL: N = new RBTreeNode<V>(NaN) as unknown as N;
override add(keyOrNode: BTNKey | N | null | undefined, value?: V): N | undefined {
let node: N;
@ -89,7 +89,7 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
node.parent = y;
if (y === undefined) {
this._root = node;
this._setRoot(node);
} else if (node.key < y.key) {
y.left = node;
} else {
@ -181,32 +181,6 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
return node !== this.NIL && node !== undefined;
}
/**
* The function `getNode` is a recursive depth-first search algorithm that searches for a node with a
* given key in a red-black tree.
* @param {number} key - The key parameter is a number that represents the value we are searching for
* in the RBTree.
* @param beginRoot - The `beginRoot` parameter is an optional parameter that represents the starting
* point for the search in the binary search tree. If no value is provided for `beginRoot`, it
* defaults to the root of the binary search tree (`this.root`).
* @returns a RBTreeNode.
*/
// getNode(key: number, beginRoot = this.root): N | undefined {
// const dfs = (node: N): N | undefined => {
// if (this.isRealNode(node)) {
// if (key === node.key) {
// return node;
// }
//
// if (key < node.key) return dfs(node.left!);
// return dfs(node.right!);
// } else {
// return undefined;
// }
// };
// return dfs(beginRoot);
// }
getNode<C extends BTNCallback<N, BTNKey>>(
identifier: BTNKey,
callback?: C,
@ -317,68 +291,16 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
return y!;
}
clear() {
override clear() {
this._root = this.NIL;
this._size = 0;
}
print(beginRoot: N = this.root) {
const display = (root: N | undefined): void => {
const [lines, , ,] = _displayAux(root);
for (const line of lines) {
console.log(line);
}
};
const _displayAux = (node: N | undefined): [string[], number, number, number] => {
if (node === undefined) {
return [[], 0, 0, 0];
}
if (node.right === undefined && node.left === undefined) {
const line = `${node.key}`;
const width = line.length;
const height = 1;
const middle = Math.floor(width / 2);
return [[line], width, height, middle];
}
if (node.right === undefined) {
const [lines, n, p, x] = _displayAux(node.left);
const s = `${node.key}`;
const u = s.length;
const first_line = ' '.repeat(x + 1) + '_'.repeat(n - x - 1) + s;
const second_line = ' '.repeat(x) + '/' + ' '.repeat(n - x - 1 + u);
const shifted_lines = lines.map(line => line + ' '.repeat(u));
return [[first_line, second_line, ...shifted_lines], n + u, p + 2, n + Math.floor(u / 2)];
}
if (node.left === undefined) {
const [lines, n, p, u] = _displayAux(node.right);
const s = `${node.key}`;
const x = s.length;
const first_line = s + '_'.repeat(x) + ' '.repeat(n - x);
const second_line = ' '.repeat(u + x) + '\\' + ' '.repeat(n - x - 1);
const shifted_lines = lines.map(line => ' '.repeat(u) + line);
return [[first_line, second_line, ...shifted_lines], n + x, p + 2, Math.floor(u / 2)];
}
const [left, n, p, x] = _displayAux(node.left);
const [right, m, q, y] = _displayAux(node.right);
const s = `${node.key}`;
const u = s.length;
const first_line = ' '.repeat(x + 1) + '_'.repeat(n - x - 1) + s + '_'.repeat(y) + ' '.repeat(m - y);
const second_line = ' '.repeat(x) + '/' + ' '.repeat(n - x - 1 + u + y) + '\\' + ' '.repeat(m - y - 1);
if (p < q) {
left.push(...new Array(q - p).fill(' '.repeat(n)));
} else if (q < p) {
right.push(...new Array(p - q).fill(' '.repeat(m)));
}
const zipped_lines = left.map((a, i) => a + ' '.repeat(u) + right[i]);
return [[first_line, second_line, ...zipped_lines], n + m + u, Math.max(p, q) + 2, n + Math.floor(u / 2)];
};
display(beginRoot);
protected override _setRoot(v: N) {
if (v) {
v.parent = undefined;
}
this._root = v;
}
/**
@ -394,7 +316,7 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
}
y.parent = x.parent;
if (x.parent === undefined) {
this._root = y;
this._setRoot(y);
} else if (x === x.parent.left) {
x.parent.left = y;
} else {
@ -419,7 +341,7 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
}
y.parent = x.parent;
if (x.parent === undefined) {
this._root = y;
this._setRoot(y);
} else if (x === x.parent.right) {
x.parent.right = y;
} else {
@ -502,7 +424,7 @@ export class RedBlackTree<V = any, N extends RBTreeNode<V, N> = RBTreeNode<V, RB
*/
protected _rbTransplant(u: N, v: N): void {
if (u.parent === undefined) {
this._root = v;
this._setRoot(v);
} else if (u === u.parent.left) {
u.parent.left = v;
} else {

14
test/unit/data-structures/binary-tree/rb-tree.test.ts
View file

@ -1,4 +1,4 @@
import {RBTNColor, RBTreeNode, RedBlackTree} from '../../../../src';
import {IterationType, RBTNColor, RBTreeNode, RedBlackTree} from '../../../../src';
import {getRandomInt} from '../../../utils';
import {isDebugTest} from '../../../config';
@ -197,7 +197,7 @@ describe('RedBlackTree', () => {
const node = tree.getNode(20);
tree.add(25);
// Verify that rotation has occurred
expect(node?.left?.key).toBe(0);
expect(node?.left?.key).toBeNaN();
expect(node?.right?.key).toBe(25);
});
@ -425,18 +425,22 @@ describe('RedBlackTree', () => {
tree.add(110);
isDebug && tree.print();
// console.log(tree.dfs())
// console.log(tree.isBST())
});
it('should fix the tree after insertion and deletion', () => {
for (let i = 0; i < 100000; i++) {
for (let i = 0; i < 1000; i++) {
tree.add(getRandomInt(-100, 1000));
}
for (let i = 0; i < 100000; i++) {
for (let i = 0; i < 1000; i++) {
tree.delete(getRandomInt(-100, 1000));
}
for (let i = 0; i < 100000; i++) {
for (let i = 0; i < 1000; i++) {
tree.add(getRandomInt(-100, 1000));
tree.delete(getRandomInt(-100, 1000));
}
// console.log(tree.dfs( n => n.key, "in", tree.root, IterationType.RECURSIVE))
});
});