mirror of
https://github.com/trekhleb/javascript-algorithms.git
synced 2026-03-13 08:51:02 +08:00
Add binary search tree.
This commit is contained in:
@@ -51,4 +51,53 @@ describe('BinaryTreeNode', () => {
|
||||
|
||||
expect(rootNode.toString()).toBe('1,2,3');
|
||||
});
|
||||
|
||||
it('should remove child node', () => {
|
||||
const leftNode = new BinaryTreeNode(1);
|
||||
const rightNode = new BinaryTreeNode(3);
|
||||
const rootNode = new BinaryTreeNode(2);
|
||||
|
||||
rootNode
|
||||
.setLeft(leftNode)
|
||||
.setRight(rightNode);
|
||||
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 3]);
|
||||
|
||||
expect(rootNode.removeChild(rootNode.left)).toBeTruthy();
|
||||
expect(rootNode.traverseInOrder()).toEqual([2, 3]);
|
||||
|
||||
expect(rootNode.removeChild(rootNode.right)).toBeTruthy();
|
||||
expect(rootNode.traverseInOrder()).toEqual([2]);
|
||||
|
||||
expect(rootNode.removeChild(rootNode.right)).toBeFalsy();
|
||||
expect(rootNode.traverseInOrder()).toEqual([2]);
|
||||
});
|
||||
|
||||
it('should replace child node', () => {
|
||||
const leftNode = new BinaryTreeNode(1);
|
||||
const rightNode = new BinaryTreeNode(3);
|
||||
const rootNode = new BinaryTreeNode(2);
|
||||
|
||||
rootNode
|
||||
.setLeft(leftNode)
|
||||
.setRight(rightNode);
|
||||
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 3]);
|
||||
|
||||
const replacementNode = new BinaryTreeNode(5);
|
||||
rightNode.setRight(replacementNode);
|
||||
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 3, 5]);
|
||||
|
||||
expect(rootNode.replaceChild(rootNode.right, rootNode.right.right)).toBeTruthy();
|
||||
expect(rootNode.right.value).toBe(5);
|
||||
expect(rootNode.right.right).toBeNull();
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 5]);
|
||||
|
||||
expect(rootNode.replaceChild(rootNode.right, rootNode.right.right)).toBeFalsy();
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 5]);
|
||||
|
||||
expect(rootNode.replaceChild(rootNode.right, replacementNode)).toBeTruthy();
|
||||
expect(rootNode.traverseInOrder()).toEqual([1, 2, 5]);
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user