|
| 1 | +/** |
| 2 | + * Definition for a binary tree node. |
| 3 | + * function TreeNode(val, left, right) { |
| 4 | + * this.val = (val===undefined ? 0 : val) |
| 5 | + * this.left = (left===undefined ? null : left) |
| 6 | + * this.right = (right===undefined ? null : right) |
| 7 | + * } |
| 8 | + */ |
| 9 | +/** |
| 10 | + * @param {TreeNode} root |
| 11 | + * @return {number} |
| 12 | + * Time O(n) where n is the number of nodes |
| 13 | + * Space O(h) where h is the hight of the tree |
| 14 | + */ |
| 15 | + |
| 16 | +// We create a class to keep track of the minimum height difference and the previous node |
| 17 | +class TreeNodeInfo { |
| 18 | + constructor(minHeightDiff, prevNode) { |
| 19 | + this.minHeightDiff = minHeightDiff; |
| 20 | + this.prevNode = prevNode; |
| 21 | + } |
| 22 | +} |
| 23 | + |
| 24 | +// This function returns the minimum height difference of a binary search tree |
| 25 | +function getMinimumHeightDiff(root) { |
| 26 | + const treeNodeInfo = new TreeNodeInfo(Infinity, null); // Initialize minimum height difference to be infinity and previous node to be null |
| 27 | + traverseInOrder(root, treeNodeInfo); // Start traversing the tree in order |
| 28 | + return treeNodeInfo.minHeightDiff; // Return minimum height difference after traversing the tree |
| 29 | +} |
| 30 | + |
| 31 | +// This function traverses a binary search tree in order and updates the minimum height difference and previous node information |
| 32 | +function traverseInOrder(node, treeNodeInfo) { |
| 33 | + if (node === null) return; // Base case |
| 34 | + |
| 35 | + traverseInOrder(node.left, treeNodeInfo); // Traverse left subtree |
| 36 | + |
| 37 | + if (treeNodeInfo.prevNode !== null) { |
| 38 | + // If previous node is not null, update the minimum height difference |
| 39 | + treeNodeInfo.minHeightDiff = Math.min( |
| 40 | + treeNodeInfo.minHeightDiff, |
| 41 | + node.val - treeNodeInfo.prevNode.val |
| 42 | + ); |
| 43 | + } |
| 44 | + |
| 45 | + treeNodeInfo.prevNode = node; // Update previous node to be the current node |
| 46 | + |
| 47 | + traverseInOrder(node.right, treeNodeInfo); // Traverse right subtree |
| 48 | +} |
0 commit comments