hxx/javascript-algorithms
1
0
Fork 0
mirror of https://github.moeyy.xyz/https://github.com/trekhleb/javascript-algorithms.git synced 2024-09-20 07:43:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enhance BinaryTreeNode: Add type checking for child nodes

Browse Source
This commit is contained in:
Anastasios Pantzartzis 2024-08-22 00:32:24 +03:00
parent ca3d16dcce
commit fc5e0d09c3
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/data-structures/tree/BinaryTreeNode.js
View File

@ -100,6 +100,12 @@ export default class BinaryTreeNode {
* @return {BinaryTreeNode}
*/
setLeft(node) {
// Check if it is a proper node.
if (node && !(node instanceof BinaryTreeNode)) {
throw new Error('The left node must be an instance of BinaryTreeNode');
}
// Reset parent for left node since it is going to be detached.
if (this.left) {
this.left.parent = null;
@ -121,6 +127,12 @@ export default class BinaryTreeNode {
* @return {BinaryTreeNode}
*/
setRight(node) {
// Check if it is a proper node.
if (node && !(node instanceof BinaryTreeNode)) {
throw new Error('The right node must be an instance of BinaryTreeNode');
}
// Reset parent for right node since it is going to be detached.
if (this.right) {
this.right.parent = null;