diff --git a/lib/tree.rb b/lib/tree.rb index c0d4b51..5deff2b 100644 --- a/lib/tree.rb +++ b/lib/tree.rb @@ -16,40 +16,158 @@ def initialize @root = nil end - # Time Complexity: - # Space Complexity: + # Time Complexity: log(n) + # Space Complexity: log(n) def add(key, value) - raise NotImplementedError + new_node = TreeNode.new(key, value) + + if @root.nil? + @root = new_node + else + add_helper(@root, new_node, key) + end end - # Time Complexity: - # Space Complexity: + def add_helper(current, new_node, key) + return new_node if current.nil? + + if key <= current.key + current.left = add_helper(current.left, new_node, key) + else + current.right = add_helper(current.right, new_node, key) + end + return current + end + + + # Time Complexity: log(n) + # Space Complexity: log(n) def find(key) - raise NotImplementedError + if @root.nil? + return nil + else + find_helper(key, @root) + end end - # Time Complexity: - # Space Complexity: + def find_helper(key, current) + return nil if current.nil? + if current.key == key + return current.value + elsif key > current.key + find_helper(key, current.right) + else + find_helper(key, current.left) + end + end + + def find_node(key) + if @root.nil? + return nil + else + find_node_helper(key, @root) + end + end + + def find_node_helper(key, current) + return nil if current.nil? + if current.key == key + return current + elsif key > current.key + find_node_helper(key, current.right) + else + find_node_helper(key, current.left) + end + end + + def delete(key) + node = find_node(key) + unless node.nil? + remove(node) + end + end + + def remove(node) + if node.left.nil? && node.right.nil? + node = nil + elsif !node.left.nil? && node.right.nil? + node = node.left + elsif node.left.nil? && !node.right.nil? + node = node.right + puts node.value + else + min_node = find_min_node(node.right) + node = min_node + min_node = nil + end + end + + def find_min_node(node) + if node.left.nil? + min_node = node + return min_node + else + find_min_node(node.left) + end + end + + # Time Complexity: O(n) + # Space Complexity: O(n) def inorder - raise NotImplementedError + return inorder_helper(@root, []) end - # Time Complexity: - # Space Complexity: + def inorder_helper(current, list) + return list if current.nil? + + inorder_helper(current.left, list) + list << { key: current.key, value: current.value} + inorder_helper(current.right, list) + + return list + end + + # Time Complexity: O(n) + # Space Complexity: O(n) def preorder - raise NotImplementedError + return preorder_helper(@root, []) end - # Time Complexity: - # Space Complexity: + def preorder_helper(node, list) + return list if node.nil? + + list << { key: node.key, value: node.value} + preorder_helper(node.left, list) + preorder_helper(node.right, list) + end + + # Time Complexity: O(n) + # Space Complexity: O(n) def postorder - raise NotImplementedError + return postorder_helper(@root, []) end - # Time Complexity: - # Space Complexity: + def postorder_helper(node, list) + return list if node.nil? + + postorder_helper(node.left, list) + postorder_helper(node.right, list) + list << { key: node.key, value: node.value} + end + + # Time Complexity: O(n) + # Space Complexity: O(log n) def height - raise NotImplementedError + return height_helper(@root) + end + + def height_helper(current) + return 0 if current.nil? + + left_height = height_helper(current.left) + right_height = height_helper(current.right) + + return [left_height, right_height].max + 1 end # Optional Method