-
-
Notifications
You must be signed in to change notification settings - Fork 223
Expand file tree
/
Copy pathbinary_search_tree.js
More file actions
136 lines (123 loc) · 3.43 KB
/
binary_search_tree.js
File metadata and controls
136 lines (123 loc) · 3.43 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
class Node {
constructor(value) {
this.left = null;
this.right = null;
this.value = value;
}
}
class BinarySearchTree {
constructor() {
this.root = null;
}
insert(value) {
const newNode = new Node(value);
if (!this.root) {
this.root = newNode;
return;
}
let currentNode = this.root;
while (true) {
if (value < currentNode.value) {
// Left
if (!currentNode.left) {
currentNode.left = newNode;
break;
}
currentNode = currentNode.left;
} else if (value > currentNode.value) {
// Right
if (!currentNode.right) {
currentNode.right = newNode;
break;
}
currentNode = currentNode.right;
} else {
break; // Duplicate value, do nothing
}
}
}
lookup(value) {
let currentNode = this.root;
while (currentNode) {
if (value < currentNode.value) {
currentNode = currentNode.left;
} else if (value > currentNode.value) {
currentNode = currentNode.right;
} else {
return true;
}
}
return false;
}
minValueNode(currentNode) {
while (currentNode && currentNode.left) {
currentNode = currentNode.left;
}
return currentNode;
}
maxValueNode(currentNode) {
while (currentNode && currentNode.right) {
currentNode = currentNode.right;
}
return currentNode;
}
bfs() {
if (!this.root) return [];
let queue = [this.root];
let output = [];
while (queue.length) {
let currentNode = queue.shift();
output.push(currentNode.value);
if (currentNode.left) queue.push(currentNode.left);
if (currentNode.right) queue.push(currentNode.right);
}
return output;
}
dfsPreOrder() {
let output = [];
function traverse(node) {
if (!node) return;
output.push(node.value);
traverse(node.left);
traverse(node.right);
}
traverse(this.root);
return output;
}
dfsPostOrder() {
let output = [];
function traverse(node) {
if (!node) return;
traverse(node.left);
traverse(node.right);
output.push(node.value);
}
traverse(this.root);
return output;
}
dfsInOrder() {
let output = [];
function traverse(node) {
if (!node) return;
traverse(node.left);
output.push(node.value);
traverse(node.right);
}
traverse(this.root);
return output;
}
}
// Usage example
const tree = new BinarySearchTree();
[20, 15, 25, 12, 16, 22, 30, 10, 14, 28, 35].forEach(v => tree.insert(v));
console.dir(tree, { depth: null });
console.log(tree.lookup(25));
console.log(tree.lookup(7));
console.log(tree.lookup(10));
console.log(tree.lookup(12));
console.log(tree.minValueNode(tree.root.left));
console.log(tree.maxValueNode(tree.root.right));
console.log(tree.bfs());
console.log(tree.dfsPreOrder());
console.log(tree.dfsPostOrder());
console.log(tree.dfsInOrder());