diff options
Diffstat (limited to 'lib/data_structures/binary_tree.rb')
| -rw-r--r-- | lib/data_structures/binary_tree.rb | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/data_structures/binary_tree.rb b/lib/data_structures/binary_tree.rb index 15b72b7..b658515 100644 --- a/lib/data_structures/binary_tree.rb +++ b/lib/data_structures/binary_tree.rb @@ -1,3 +1,5 @@ +require_relative "../utility/block_visitor" + class BinaryTree def push(item) if @root @@ -17,6 +19,10 @@ class BinaryTree @root.accept(visitor, traversal) if @root end + def each(&block) + accept(BlockVisitor.new(&block)) + end + class BinaryTreeNode attr_reader :left, :right, :data |