Diana - Space

lib/tree.rb

@@ -2,54 +2,105 @@ class TreeNode
   attr_reader :key, :value
   attr_accessor :left, :right
 
-   def initialize(key, val)
-    @key = key
-    @value = val
-    @left = nil
-    @right = nil
-   end
-end
-
-class Tree
-  attr_reader :root
-  def initialize
-    @root = nil
+    def initialize(key, val)
+      @key = key
+      @value = val
+      @left = nil
+      @right = nil
+    end
   end
+
+  class Tree
+    attr_reader :root
+    def initialize
+      @root = nil
+    end
+
+  # Time Complexity: worst case unbalanced-O(n), bal-O(log n)
+  # Space Complexity: worst case-O(n), bal-O(log n)
+    def add(key, value)
+      if @root.nil?
+        @root = TreeNode.new(key, value)
+      else
+        self.add_helper(@root, key, value)
+      end
+    end
 
-  # Time Complexity: 
-  # Space Complexity: 
-  def add(key, value)
-    raise NotImplementedError
-  end
+    def add_helper(node, new_key, new_value)
+      if new_key <= node.key
+        node.left.nil? ? node.left = TreeNode.new(new_key, new_value) : add_helper(node.left, new_key, new_value)
+      else
+        node.right.nil? ? node.right = TreeNode.new(new_key, new_value) : add_helper(node.right, new_key, new_value)
+      end
+    end
+
+  # Time Complexity: unbal-O(n), bal-O(log n)
+  # Space Complexity: unbal-O(n), bal-O(log n)
+  def find(key, node=@root)
+    return nil if @root.nil?
 
-  # Time Complexity: 
-  # Space Complexity: 
-  def find(key)
-    raise NotImplementedError
+    if key < node.key
+      find(key, node.left)
+    elsif key > node.key
+      find(key, node.right)
+    else
+      return node.value
+    end
   end
 
-  # Time Complexity: 
-  # Space Complexity: 
+
+  # Time Complexity: O(n)
+  # Space Complexity: O(n)
   def inorder
-    raise NotImplementedError
+    return inorder_helper(@root, [])
   end
 
-  # Time Complexity: 
-  # Space Complexity: 
+  def inorder_helper(node, array)
+    return array if node.nil?
+
+    inorder_helper(node.left, array)
+    array << {:key=>node.key, :value=>node.value}
+    inorder_helper(node.right, array)
+
+    return array
+  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, array)
+    return [] if node.nil?
+
+    array << {:key=>node.key, :value=>node.value}
+    preorder_helper(node.left, array)
+    preorder_helper(node.right, array)
+
+    return array
+  end
+
+  # Time Complexity: O(n)
+  # Space Complexity: O(n)
   def postorder
-    raise NotImplementedError
+    return postorder_helper(@root, [])
   end
 
-  # Time Complexity: 
-  # Space Complexity: 
-  def height
-    raise NotImplementedError
+  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}
+
+    return array
+  end
+
+  # Time Complexity: O(n)
+  # Space Complexity: O(1)
+  def height(node=@root)
+    node.nil? ? 0 : (1 + [height(node.right), height(node.left)].max)
   end
 
   # Optional Method