2011-01-07 11 views
25

Klasa A posiada następujące porównanie:Jak przekazać niestandardowy komparator do "sortowania"?

class A 
    attr_accessor x 

    def my_comparator(a) 
    x**2 <=> (a.x)**2 
    end 
end 

chciałbym używać tego komparatora posortować tablicę gdzie każdy element jest klasy A:

class B 
    def my_method 
    items.sort!(<how can I pass my_comparator here ?>) 
    end 
end 

Jak mam przejść my_comparator do sort!?

+1

Dosłownie można użyć 'items.sort! {| x, y | x.my_comparator y} ', ale jeśli jest to domyślne zachowanie sortowania dla klasy, powinieneś rozważyć coś takiego jak to, co Tin Man ma poniżej. – coreyward

Odpowiedz

33

definiować własne <=> i obejmują porównywalne. To od Comparable doc:

class SizeMatters 
    include Comparable 
    attr :str 
    def <=>(anOther) 
    str.size <=> anOther.str.size 
    end 
    def initialize(str) 
    @str = str 
    end 
    def inspect 
    @str 
    end 
end 

s1 = SizeMatters.new("Z") 
s2 = SizeMatters.new("YY") 
s3 = SizeMatters.new("XXX") 
s4 = SizeMatters.new("WWWW") 
s5 = SizeMatters.new("VVVVV") 

s1 < s2      #=> true 
s4.between?(s1, s3)   #=> false 
s4.between?(s3, s5)   #=> true 
[ s3, s2, s5, s4, s1 ].sort #=> [Z, YY, XXX, WWWW, VVVVV] 

rzeczywistości nie muszą zawierać porównywalne, ale masz dodatkowe funkcje za darmo, jeśli robisz to po zdefiniowane <=>.

W przeciwnym razie można użyć bloku Enumerable's sort z blokiem, jeśli obiekty już implementują <=>.

EDYCJA: Innym sposobem użycia kilku różnych porównań jest użycie lambdas. Wykorzystuje nową składnię 1.9.2 deklarację:

ascending_sort = ->(a,b) { a <=> b } 
descending_sort = ->(a,b) { b <=> a } 

[1, 3, 2, 4].sort(& ascending_sort) # => [1, 2, 3, 4] 
[1, 3, 2, 4].sort(& descending_sort) # => [4, 3, 2, 1] 

foo = ascending_sort 
[1, 3, 2, 4].sort(& foo) # => [1, 2, 3, 4] 
+1

Lub, specyficzne dla tego pytania: 'alias_method: <=>,: my_comparator' – Phrogz

+0

+1. Dobry chwyt @Phrogz, chociaż byłoby bardziej Ruby-owskie nazywanie metody '<=>' w pierwszej kolejności. –

+0

W moim przypadku mam kilka komparatorów, więc nie chcę przesłonić '<=>' z 'my_comparator'. –

16

Obie te powinny działać:

items.sort_by! { |a| (a.x)**2 } 
items.sort! { |a1,a2| a1.my_comparator(a2) } 
+0

To jest dobre i poprawne, ale odpowiedź @TinMan jest lepsza dla klas niestandardowych. – Phrogz

+0

To była odpowiedź, której szukałem, a nie to, że tam jest Bzdura. –

5
items.sort!(&:my_comparator) 

ten wywołuje :my_comparator.to_proc wewnętrznie, która zwraca blok

proc {|x,y| x.my_comparator(y)} 

zmniejszając tę ​​odpowiedź za odpowiedź Ben Alpert.

(Ale zgadzam się z obserwacji Phrogz, że jeśli jest to naturalny zamówienie dla klasy, należy użyć odpowiedź Blaszany mężczyzny zamiast.)

Powiązane problemy