forked from knaxus/problem-solving-javascript
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
39 lines (33 loc) · 876 Bytes
/
index.js
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
// eslint-disable-next-line no-unused-vars
const BST = require('../index');
// Inorder traversal returns a sorted array
function inOrderTraversal(root) {
if (root === null) return [];
let arr = [];
// traverse left
const left = inOrderTraversal(root.leftChild);
arr = [...left, root.value];
const right = inOrderTraversal(root.rightChild);
return [...arr, ...right];
}
function findKthMax(rootNode, k) {
const arr = inOrderTraversal(rootNode);
if (k <= 0 || k > arr.lenth) {
throw new Error('Invalid value for K');
}
return arr[arr.length - k];
}
// // create a BST
// const myBST = new BST(6);
// myBST.add(2);
// myBST.add(19);
// myBST.add(14);
// myBST.add(8);
// myBST.add(5);
// myBST.add(12);
// myBST.add(33);
// myBST.add(52);
// myBST.add(1);
// // find 3rd max
// console.log(findKthMax(myBST.root, 3));
module.exports = findKthMax;