0

我对 Comparator 接口有一个误解，它的方法 compare 这里是下面的代码，我想知道为什么 compare 方法返回 -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
} }
4

2 回答 2

4

你可能知道 if a-b=cthen b-a=-c。

这里发生的事情非常相似。您似乎已经假设它TreeSet调用了compare这样的方法：

comparator.compare(s1, s2)

（请注意，我使用s1ands2用于演示目的。它们显然不在 . 的范围内，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了我描述的方式。

编辑：

正如 Holger 在评论中所说，您不应该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));
于 2019-02-16T10:37:38.450 回答
0

中的 s1 和 s2compare(Sorted s1, Sorted s2)是局部变量定义，您不能将它们与 中的定义混淆main()。它没有定义（算法上，仅由实现）如何TreeSet比较这两个元素。

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

TreeSet内部使用 a TreeMap。put在几个地方调用 compare，通常将您放入 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;
    }

其他实现或方法可能具有不同的行为。

于 2019-02-16T10:42:01.450 回答