From 6e866007b65d55d41b037054960a7a02afc88510 Mon Sep 17 00:00:00 2001
From: Katie <katie.vandervoort@gmail.com>
Date: Tue, 1 Sep 2020 00:42:27 -0700
Subject: [PATCH] implemented all methods to pass all tests

---
 lib/tree.rb | 107 ++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 99 insertions(+), 8 deletions(-)

diff --git a/lib/tree.rb b/lib/tree.rb
index c0d4b51..867f3b6 100644
--- a/lib/tree.rb
+++ b/lib/tree.rb
@@ -19,48 +19,139 @@ def initialize
   # Time Complexity: 
   # Space Complexity: 
   def add(key, value)
-    raise NotImplementedError
+    new_node = TreeNode.new(key, value)
+    if @root == nil
+      @root = new_node
+      return new_node
+    else
+      add_helper(@root, new_node)
+    end
+  end
+
+  def add_helper(current, new_node)
+    if new_node.key <= current.key
+      if current.left == nil
+        return current.left = new_node
+      else
+        add_helper(current.left, new_node)
+      end
+    else
+      if current.right == nil
+        return current.right = new_node
+      else
+        add_helper(current.right, new_node)
+      end
+    end 
   end
 
   # Time Complexity: 
   # Space Complexity: 
   def find(key)
-    raise NotImplementedError
+    if @root == nil
+      return nil
+    elsif @root.key == key
+      return @root.value
+    else
+      current_node = @root
+      while current_node != nil
+        if key == current_node.key
+          return current_node.value
+        elsif key <= current_node.key
+          current_node = current_node.left
+        else
+          current_node = current_node.right
+        end
+      end
+    end  
   end
 
+
+
   # Time Complexity: 
   # Space Complexity: 
   def inorder
-    raise NotImplementedError
+    ordered_array = []
+    if @root != nil
+      inorder_helper(@root, ordered_array)
+    end
+    return ordered_array
+  end
+
+  def inorder_helper(node, array)
+    return if node.nil?
+    inorder_helper(node.left, array)
+    array.push({"key": node.key, "value": node.value})
+    inorder_helper(node.right, array)
   end
 
   # Time Complexity: 
   # Space Complexity: 
   def preorder
-    raise NotImplementedError
+    preordred_array = []
+    if @root != nil
+      preorder_helper(@root, preordred_array)
+    end
+    return preordred_array
   end
 
+  def preorder_helper(node, array)
+    return if node.nil?
+    array << {key: node.key, value: node.value}
+
+    preorder_helper(node.left, array)
+    preorder_helper(node.right, array)
+  end
+
+
+
+
   # Time Complexity: 
   # Space Complexity: 
   def postorder
-    raise NotImplementedError
+    preordred_array = []
+    if @root != nil
+      postorder_helper(@root, preordred_array)
+    end
+    return preordred_array
+  end
+
+  def postorder_helper(node, array)
+    return if node.nil?
+
+    postorder_helper(node.left, array)
+    postorder_helper(node.right, array)
+    array << {key: node.key, value: node.value}
   end
 
   # Time Complexity: 
   # Space Complexity: 
   def height
-    raise NotImplementedError
+    count = 0
+    if @root != nil
+      count = height_helper(@root)
+    end
+    return count
+  end
+
+  def height_helper(root)
+    return 0 if root.nil?
+
+    return find_max(height_helper(root.left), height_helper(root.right)) + 1
+  end
+
+  def find_max(a,b)
+    a >= b ? a : b
   end
 
   # Optional Method
   # Time Complexity: 
   # Space Complexity: 
   def bfs
-    raise NotImplementedError
+    # raise NotImplementedError
   end
 
   # Useful for printing
   def to_s
-    return "#{self.inorder}"
+    # return "#{self.inorder}"
   end
 end