Class: Intervals::Tree

Inherits:
Object
  • Object
show all
Defined in:
lib/itree/tree.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Tree

Returns a new instance of Tree.



5
6
7
8
 
# File 'lib/itree/tree.rb', line 5

def initialize
	@root = nil
	@size = 0
end

Instance Attribute Details

#root ⇒ Object

Returns the value of attribute root.



3
4
5
 
# File 'lib/itree/tree.rb', line 3

def root
  @root
end

#size ⇒ Object

Returns the value of attribute size.



3
4
5
 
# File 'lib/itree/tree.rb', line 3

def size
  @size
end

Instance Method Details

#insert(leftScore, rightScore, data = nil, updateData = false) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/itree/tree.rb', line 10

def insert(leftScore, rightScore, data=nil, updateData=false)
	node = Node.new(leftScore,rightScore,data)
	success = true

	if @root.nil?
		@root = node
	else
		balance,success = insertNode(@root,node,updateData)
	end

	@size = @size + 1 if success
	return success
end

#insert!(leftScore, rightScore, data = nil) ⇒ Object 



24
25
26
 
# File 'lib/itree/tree.rb', line 24

def insert!(leftScore, rightScore, data=nil)
	return insert(leftScore,rightScore,data,true)
end

#remove(leftScore, rightScore) ⇒ Object 



35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/itree/tree.rb', line 35

def remove(leftScore, rightScore)
	if @root
		delNode = Node.new(leftScore,rightScore,nil)
		height,removed = removeNode(@root,delNode)
		if removed
			@size = @size - 1
			@root = nil if @size == 0
			return true
		end
	end
	return false
end

#stab(minScore, maxScore = nil) ⇒ Object 



28
29
30
31
32
33
 
# File 'lib/itree/tree.rb', line 28

def stab(minScore,maxScore=nil)
	maxScore = minScore if maxScore.nil?
	results = []
	stabNode(@root,minScore,maxScore,results) if @root
	results
end