diff --git a/lib/rbtree_map.rb b/lib/rbtree_map.rb index fc2a828..9053f18 100644 --- a/lib/rbtree_map.rb +++ b/lib/rbtree_map.rb @@ -1,7 +1,7 @@ class RBTreeMap class RBTree - attr_reader :key, :value, :color - attr_reader :left, :right + attr_reader :key, :value + attr_accessor :color, :left, :right def initialize(key, value) @key, @value = key, value @@ -87,18 +87,6 @@ def need_rebalance? red? and (@right.red? or @left.red?) end - def color=(color) - @color = color - end - - def left=(left) - @left = left - end - - def right=(right) - @right = right - end - def color_flip(other) @color, other.color = other.color, @color end