Интерфейс Java Comparator - PullRequest
       35

Интерфейс Java Comparator

0 голосов
/ 16 февраля 2019

У меня неправильное представление об интерфейсе Comparator, и его метод сравнения приведен в следующем коде, и мне интересно, зачем сравнивать метод return -33 Я считаю, что он должен возвращать 33

import java.util.*;
public class Sorted implements Comparable<Sorted>, Comparator<Sorted> {
private int num;
private String text;

Sorted(int n, String t) {
    this.num = n;
    this.text = t;
}

public String toString() {
    return "" + num;
}

public int compareTo(Sorted s) {
    return text.compareTo(s.text);
}

public int compare(Sorted s1, Sorted s2) {
    System.out.println(s1.num-s2.num); // return -33
    return s1.num - s2.num;
}
public static void main(String[] args) {
    Sorted s1 = new Sorted(88, "a");
    Sorted s2 = new Sorted(55, "b");
    TreeSet<Sorted> t1 = new TreeSet<>();
    t1.add(s1); t1.add(s2);
    TreeSet<Sorted> t2 = new TreeSet<>(s1);
    t2.add(s1); t2.add(s2);
    System.out.println(t1 + " " + t2);
    System.out.println(s1.num-s2.num); // prints 33
} }

Ответы [ 2 ]

0 голосов
/ 16 февраля 2019

s1 и s2 в compare(Sorted s1, Sorted s2) являются определениями локальных переменных, их не следует путать с определениями в main().Не определено (алгоритмически, только реализацией), как TreeSet сравнивает два элемента.

compare(s1, s2) //yields 33
compare(s2, s1) //yields -33

TreeSet внутренне использует TreeMap.put вызовы сравниваются в нескольких местах, обычно с элементом, который вы помещаете в TreeSet в качестве первого элемента.Поэтому put(s2) назовет compare(s2, s1).См. Фрагмент кода ниже:

public V put(K key, V value) {    
        Entry<K,V> t = root;    
        if (t == null) {
            compare(key, key); // type (and possibly null) check       
            root = new Entry<>(key, value, null);  
            size = 1;
            modCount++;    
            return null;    
        }    
        int cmp;    
        Entry<K,V> parent;    
        // split comparator and comparable paths    
        Comparator<? super K> cpr = comparator;    
        if (cpr != null) {    
            do {
                parent = t;    
                cmp = cpr.compare(key, t.key);    
                if (cmp < 0)   
                    t = t.left;    
                else if (cmp > 0)    
                    t = t.right;    
                else    
                    return t.setValue(value);    
            } while (t != null);    
        }    
        else {   
            if (key == null)    
                throw new NullPointerException();    
            @SuppressWarnings("unchecked")    
                Comparable<? super K> k = (Comparable<? super K>) key;   
            do {    
                parent = t;    
                cmp = k.compareTo(t.key);    
                if (cmp < 0)
                     t = t.left; 
                else if (cmp > 0)
                    t = t.right;
                else
                    return t.setValue(value);    
            } while (t != null);    
        }    
        Entry<K,V> e = new Entry<>(key, value, parent);    
        if (cmp < 0)
                parent.left = e;    
        else    
            parent.right = e;    
        fixAfterInsertion(e);    
        size++;    
        modCount++;

        return null;
    }

Другие реализации или методы могут иметь другое поведение.

0 голосов
/ 16 февраля 2019

Вы, наверное, знаете, что если a-b=c, то b-a=-c.

То, что здесь происходит, очень похоже.Похоже, вы предположили, что TreeSet вызывает метод compare следующим образом:

comparator.compare(s1, s2)

(Обратите внимание, что я использовал s1 и s2 в демонстрационных целях. Они явно не входят в сферу примененияв TreeSet. s1 - это тот же экземпляр, что и ваш s1, а s2 - это тот же экземпляр, что и ваш s2`.)

Но он также может вызвать compare так:

comparator.compare(s2, s1)

не так ли?

Если он называет это вторым способом, то следует ожидать результата -33.

РЕДАКТИРОВАТЬ:

Я посмотрел исходный код для TreeSet.add и обнаружил, что он вызывает TreeMap.put с элементом, который вы добавляете в качестве ключа.Если вы посмотрите дальше на TreeMap.put, вы найдете:

Comparator<? super K> cpr = comparator;
if (cpr != null) {
    do {
        parent = t;
        cmp = cpr.compare(key, t.key); // <--- "key" is the key passed into this method
                                       // "t" is an element that is already in the map
        if (cmp < 0)
            t = t.left;
        else if (cmp > 0)
            t = t.right;
        else
            return t.setValue(value);
    } while (t != null);
}

Это показывает, что TreeSet действительно вызывает compare, как я описал.

РЕДАКТИРОВАТЬ:

Как сказал Хольгер в комментариях, вы не должны реализовывать Comparator, вычитая два целых числа.Вместо этого вы должны использовать Integer.compare:

return Integer.compare(s1.num, s2.num);

На самом деле вообще не нужно реализовывать Comparator, вы можете передать Comparator.comparingInt(s -> s.num) при создании TreeMap:

TreeSet<Sorted> t1 = new TreeSet<>(Comparator.comparingInt(s -> s.num));
...