From b60700ad87cd05eb0747bf82e00fa39c75859625 Mon Sep 17 00:00:00 2001 From: Sharon Cheung Date: Sat, 29 Aug 2020 23:05:16 -0700 Subject: [PATCH 1/2] all test passed --- lib/tree.rb | 100 ++++++++++++++++++++++++++++++++++++++++++++++++---- 1 file changed, 94 insertions(+), 6 deletions(-) diff --git a/lib/tree.rb b/lib/tree.rb index c0d4b51..bbe0525 100644 --- a/lib/tree.rb +++ b/lib/tree.rb @@ -19,37 +19,125 @@ def initialize # Time Complexity: # Space Complexity: def add(key, value) - raise NotImplementedError + new_node = TreeNode.new(key, value) + + if @root.nil? + @root = new_node + else + add_helper(@root, key, value) + end + end + + def add_helper(current, key, value) + return TreeNode.new(key,value) if current.nil? + + if key < current.key + current.left = add_helper(current.left, key, value) + else + current.right = add_helper(current.right, key, value) + end + + return current end # Time Complexity: # Space Complexity: def find(key) - raise NotImplementedError + return if @root.nil? + + if @root == key + return @root.value + else + find_helper(@root, key) + end + end + + def find_helper(current, key) + return if current.nil? + + if key < current.key + current.left = find_helper(current.left, key) + elsif key > current.key + current.right = find_helper(current.right, key) + else + return current.value + end end # Time Complexity: # Space Complexity: def inorder - raise NotImplementedError + return inorder_helper(@root, []) + end + + def inorder_helper(current_node, list) + return list if current_node.nil? + # left + inorder_helper(current_node.left, list) + # middle + list << {key: current_node.key, value: current_node.value} + # right + inorder_helper(current_node.right, list) + + return list end # Time Complexity: # Space Complexity: def preorder - raise NotImplementedError + return preorder_helper(@root, []) + end + + def preorder_helper(current_node, list) + return list if current_node.nil? + # middle + list << {key: current_node.key, value: current_node.value} + # left + preorder_helper(current_node.left, list) + #right + preorder_helper(current_node.right, list) + + return list end # Time Complexity: # Space Complexity: def postorder - raise NotImplementedError + return postorder_helper(@root, []) + end + + def postorder_helper(current_node, list) + return list if current_node.nil? + # left + postorder_helper(current_node.left, list) + #right + postorder_helper(current_node.right, list) + # middle + list << {key: current_node.key, value: current_node.value} + + return list end # Time Complexity: # Space Complexity: def height - raise NotImplementedError + if @root.nil? + return 0 + else + height_helper(@root) + end + end + + def height_helper(current) + return 0 if current.nil? + left = height_helper(current.left) + right = height_helper(current.right) + + if (left > right) + return left + 1 + else + return right + 1 + end end # Optional Method From 948539fcea21207a7c364ebf338d23d177333ab5 Mon Sep 17 00:00:00 2001 From: Sharon Cheung Date: Mon, 31 Aug 2020 22:46:09 -0700 Subject: [PATCH 2/2] added complexity answers --- lib/tree.rb | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/lib/tree.rb b/lib/tree.rb index bbe0525..662347d 100644 --- a/lib/tree.rb +++ b/lib/tree.rb @@ -16,8 +16,8 @@ def initialize @root = nil end - # Time Complexity: - # Space Complexity: + # Time Complexity: O(log n) + # Space Complexity: O(log n) def add(key, value) new_node = TreeNode.new(key, value) @@ -40,8 +40,8 @@ def add_helper(current, key, value) return current end - # Time Complexity: - # Space Complexity: + # Time Complexity: O(log n) + # Space Complexity: O(log n) def find(key) return if @root.nil? @@ -64,8 +64,8 @@ def find_helper(current, key) end end - # Time Complexity: - # Space Complexity: + # Time Complexity: O(n) + # Space Complexity: O(n) def inorder return inorder_helper(@root, []) end @@ -82,8 +82,8 @@ def inorder_helper(current_node, list) return list end - # Time Complexity: - # Space Complexity: + # Time Complexity: O(n) + # Space Complexity: O(n) def preorder return preorder_helper(@root, []) end @@ -100,8 +100,8 @@ def preorder_helper(current_node, list) return list end - # Time Complexity: - # Space Complexity: + # Time Complexity: O(n) + # Space Complexity: O(n) def postorder return postorder_helper(@root, []) end