使用自定义排序功能实现 Treeset

Question

我正在尝试使用自定义排序功能实现树集。我已经传递了包含排序逻辑的比较器对象。

import java.util.Comparator;
import java.util.TreeSet;

public class MySetWithCompr {

    public static void main(String a[]){

        TreeSet<String> ts = new TreeSet<String>(new MyComp());
        ts.add("Test1");
        ts.add("Test2");
        ts.add("Test3");
        ts.add("Test4");
        System.out.println(ts);
    }
}

class MyComp implements Comparator<String>{

    @Override
    public int compare(String str1, String str2) {
        return str1.compareTo(str2);
    }

}

上面的程序给了我下面的输出

[Test1, Test2, Test3, Test4]

请建议如何实现比较器逻辑以使结果相反

Expected Result : [Test4, Test3, Test2, Test1]

Answer 1

如果您使用的是 Java 8，您可以简单地编写：

Set<String> ts = new TreeSet<>(new MyComp().reversed());

在 Java 8 之前（Java 5 起）有相同的功能：

Set<String> ts = new TreeSet<String>(Collections.reverseOrder(new MyComp()));

Answer 2

试试看：

class MyComp implements Comparator<String> {

    @Override
    public int compare(String str1, String str2) {
        return str2.compareTo(str1);
    }

}

Answer 3

如果你想 desc 排序有两种方法：

第一：

class MyComp implements Comparator<String> {
    @Override
    public int compare(String str1, String str2) {
        return str2.compareTo(str1);
    }
}

第二：

class MyComp implements Comparator<String> {
    @Override
    public int compare(String str1, String str2) {
        return -1 * str1.compareTo(str2);
    } 
}