Ruby数组交集

时间:2011-10-12 17:12:36

标签: ruby arrays

使用'&'时,两个包含对象的数组不会返回相交数组之间。 请看下面的代码:

ruby-1.9.2-p290 :001 > class A
ruby-1.9.2-p290 :002?>   include Comparable
ruby-1.9.2-p290 :003?>   attr_reader :key
ruby-1.9.2-p290 :004?>   def initialize(key)
ruby-1.9.2-p290 :005?>     @key = key
ruby-1.9.2-p290 :006?>     end
ruby-1.9.2-p290 :007?>   def <=> obj
ruby-1.9.2-p290 :008?>     @key <=> obj.key
ruby-1.9.2-p290 :009?>     end
ruby-1.9.2-p290 :010?>   end
 => nil 
ruby-1.9.2-p290 :011 > class B
ruby-1.9.2-p290 :012?>   attr_reader :key
ruby-1.9.2-p290 :013?>   def initialize(key)
ruby-1.9.2-p290 :014?>     @key = key
ruby-1.9.2-p290 :015?>     end
ruby-1.9.2-p290 :016?>   end
 => nil 
ruby-1.9.2-p290 :017 > A.new(1) == A.new(1)
 => true 
ruby-1.9.2-p290 :019 > B.new(1) == B.new(1)
 => false 
ruby-1.9.2-p290 :020 > a1 = [A.new(1), A.new(2), A.new(3)]
 => [#<A:0x000001009e2f68 @key=1>, #<A:0x000001009e2f40 @key=2>, #<A:0x000001009e2f18 @key=3>] 
ruby-1.9.2-p290 :021 > a2 = [A.new(3), A.new(4), A.new(5)]
 => [#<A:0x000001009d44e0 @key=3>, #<A:0x000001009d44b8 @key=4>, #<A:0x000001009d4490 @key=5>] 
ruby-1.9.2-p290 :023 > a1 | a2
 => [#<A:0x000001009e2f68 @key=1>, #<A:0x000001009e2f40 @key=2>, #<A:0x000001009e2f18 @key=3>, #<A:0x000001009d44e0 @key=3>, #<A:0x000001009d44b8 @key=4>, #<A:0x000001009d4490 @key=5>] 
ruby-1.9.2-p290 :024 > a1 & a2
 => [] 

不应该a1&amp; a2返回:

[#<A:0x000001009e2f18 @key=3>]

或者,我只是遗漏了一些东西......

1 个答案:

答案 0 :(得分:9)

不,您需要实现Array#&Array#|的哈希等式才能工作(仅通过正常比较实现它将是O(n * m))。注意Array#|也返回了错误的结果:它包含重复项。

这种平等方法可以这样实现:

 def hash
   @key.hash ^ A.hash # just to get a different hash than the key
 end

 alias eql? ==

此外,如果其他对象没有响应#key,则<=>会失败。 ==不应该失败,如果无法比较两个对象,它应该返回false。这也是你不想使用respond_to?的方法之一,而是is_a?:你不希望电影等于一本书,因为它们恰好具有相同的标题。

def <=>(other)
  @key <=> other.key if other.is_a? A
end