Class: Kruskal::Forest

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/kruskal/forest.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Forest

Returns a new instance of Forest.



5
6
7
 
# File 'lib/kruskal/forest.rb', line 5

def initialize
  @trees = {}
end

Instance Method Details

#[](position) ⇒ Object 



26
27
28
 
# File 'lib/kruskal/forest.rb', line 26

def [](position)
  to_a[position]
end

#each ⇒ Object 



30
31
32
33
34
 
# File 'lib/kruskal/forest.rb', line 30

def each
  to_a.each do |tree|
    yield tree
  end
end

#find_tree_for(node) ⇒ Object 



22
23
24
 
# File 'lib/kruskal/forest.rb', line 22

def find_tree_for(node)
  @trees[node]
end

#index(node, tree) ⇒ Object 



9
10
11
12
 
# File 'lib/kruskal/forest.rb', line 9

def index(node, tree)
  @trees[node] = tree
  @cached_to_a = nil
end

#new_tree(value = 0, source, target) ⇒ Object 



14
15
16
 
# File 'lib/kruskal/forest.rb', line 14

def new_tree(value = 0, source, target)
  Tree.new(self, [[value, source, target]])
end

#size ⇒ Object 



36
37
38
 
# File 'lib/kruskal/forest.rb', line 36

def size
  to_a.size
end

#to_a ⇒ Object 



18
19
20
 
# File 'lib/kruskal/forest.rb', line 18

def to_a
  @cached_to_a ||= @trees.values.uniq
end