forked from tigarcia/BinarySearchTreeRuby
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathbinary_search_tree.rb
105 lines (87 loc) · 2.04 KB
/
binary_search_tree.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
require 'pry'
class BinarySearchTree
attr_reader :size
class Node
attr_accessor :value, :left, :right
def initialize(value, left=nil, right=nil)
@value = value
@left = left
@right = right
end
end
def initialize(values=[])
@root = nil
@size = 0
if values.length > 0
## TODO: Add the values from the array to the tree
end
end
def insert(value)
if @root.nil?
@root = Node.new(value)
@size += 1
else
insert_internal(value, @root)
end
self
end
def to_s()
print_internal(@root, 0)
end
def find(value)
node = find_internal(value, @root)
node.nil? ? nil : node.value
end
def delete(value)
delete_internal(value, @root)
end
private
def insert_internal(value, node)
if value <= node.value
if node.left.nil?
node.left = Node.new(value)
@size += 1
else
insert_internal(value, node.left)
end
else
if node.right.nil?
node.right = Node.new(value)
@size += 1
else
insert_internal(value, node.right)
end
end
end
def print_internal(node, tab_count)
unless node.nil?
tab_count.times {|n| print " "}
puts node.value
tab_count.times {|n| print " "}
if node.left.nil?
puts " left: nil"
else
puts " left:"
print_internal(node.left, tab_count + 1)
end
tab_count.times {|n| print " "}
if node.right.nil?
puts " right: nil"
else
puts " right:"
print_internal(node.right, tab_count + 1)
end
end
end
def find_internal(value, node)
end
def delete_internal(value, node)
end
end
#### Example usage
# tree = BinarySearchTree.new
# tree.insert("galvanize").insert("apple").insert("pie").insert("pizza").insert("pi")
# puts tree.to_s
# tree.size #should equal 5
# tree.find("apple") #should return "apple"
# tree.find("apples") # should return nil