1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

speed up set intersect

Browse Source
This commit is contained in:
Oleg Zubchenko 2018-11-03 15:27:10 +03:00 committed by Akinori MUSHA
parent 0cf75e3850
commit 4ce28b58cb
Notes: git 2019-12-31 20:53:05 +09:00 
Merged: https://github.com/ruby/ruby/pull/2003
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/set.rb
View File

@ -464,7 +464,15 @@ class Set
# Set['a', 'b', 'z'] & ['a', 'b', 'c'] #=> #<Set: {"a", "b"}> # Set['a', 'b', 'z'] & ['a', 'b', 'c'] #=> #<Set: {"a", "b"}>
def &(enum) def &(enum)
n = self.class.new n = self.class.new
if enum.is_a?(Set)
if enum.size > size
each { |o| n.add(o) if enum.include?(o) }
else
enum.each { |o| n.add(o) if include?(o) }
end
else
do_with_enum(enum) { |o| n.add(o) if include?(o) } do_with_enum(enum) { |o| n.add(o) if include?(o) }
end
n n
end end
alias intersection & alias intersection &