Java TreeSet(Comparator comparator) - 使用比较器排序

Java TreeSet(Comparator comparator) - 使用比较器排序

TreeSet(Comparator comparator) 构造方法创建一个新的空 TreeSet，它根据指定的比较器进行排序。比较器可以是自然顺序或自定义顺序。如果指定的比较器为 null，则将使用元素类型的自然顺序进行排序。

例如，以下代码创建了一个 TreeSet，它根据字符串长度进行排序：

TreeSet<String> treeSet = new TreeSet<>(new Comparator<String>() {
    @Override
    public int compare(String o1, String o2) {
        return Integer.compare(o1.length(), o2.length());
    }
});

treeSet.add('apple');
treeSet.add('banana');
treeSet.add('pear');
treeSet.add('orange');

System.out.println(treeSet); // [pear, apple, banana, orange]

在这个例子中，我们传递了一个自定义比较器，它根据字符串长度进行比较。因此，最终 TreeSet 中的元素按字符串长度升序排列。

通过 TreeSet(Comparator comparator) 构造方法，您可以灵活地定义 TreeSet 中元素的排序规则，使其满足您的特定需求。