hello-algo/en/codes/javascript/chapter_tree/binary_tree.js

36 lines
820 B
JavaScript

/**
* File: binary_tree.js
* Created Time: 2022-12-04
* Author: IsChristina (christinaxia77@foxmail.com)
*/
const { TreeNode } = require('../modules/TreeNode');
const { printTree } = require('../modules/PrintUtil');
/* Initialize binary tree */
// Initialize node
let n1 = new TreeNode(1),
n2 = new TreeNode(2),
n3 = new TreeNode(3),
n4 = new TreeNode(4),
n5 = new TreeNode(5);
// Construct node references (pointers)
n1.left = n2;
n1.right = n3;
n2.left = n4;
n2.right = n5;
console.log('\nInitialize binary tree\n');
printTree(n1);
/* Insert and remove nodes */
const P = new TreeNode(0);
// Insert node P between n1 -> n2
n1.left = P;
P.left = n2;
console.log('\nAfter inserting node P\n');
printTree(n1);
// Remove node P
n1.left = n2;
console.log('\nAfter removing node P\n');
printTree(n1);