使用 Java TreeSet 对 Pair 排序并处理重复值_技术教程

本文介绍了在使用 Java `TreeSet` 对 `Pair` 对象进行排序时，如何解决因 `Comparator` 定义导致具有相同 value 值的 `Pair` 对象无法插入的问题。通过扩展 `Comparator`，使得在 value 值相同的情况下，根据 key 值进行进一步排序，从而保证所有不同的 `Pair` 对象都能正确插入到 `TreeSet` 中。

在使用 TreeSet 对自定义对象进行排序时，Comparator 的定义至关重要。如果 Comparator 将两个不同的对象判定为相等，那么 TreeSet 将只会保留其中一个。当我们需要保留所有不同的对象，即使它们在某些属性上具有相同的值时，就需要仔细设计 Comparator。

问题分析

在使用 TreeSet 存储 Pair 对象，并按照 value 值降序排序时，如果两个 Pair 对象的 value 值相同，原始的 Comparator 会认为它们相等。这导致后续插入具有相同 value 但不同 key 的 Pair 对象时，TreeSet 会拒绝插入，因为它已经包含一个“相等”的对象。

解决方案

为了解决这个问题，我们需要扩展 Comparator，使其在 value 值相同的情况下，进一步比较 key 值。这样，只有当两个 Pair 对象的 value 和 key 都相同时，才会被认为是相等的。

一种可行的方案是：

首先按照 value 值降序排序。
然后按照 key 值升序（或降序）排序。

以下是使用 Java 8 的 Comparator.comparingInt 和 thenComparing 方法实现的示例代码：

import javafx.util.Pair;

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

public class TreeSetPairExample {

    public static void main(String[] args) {
        // 定义一个 TreeSet，使用扩展的 Comparator
        final TreeSet> sortedSet = new TreeSet<>(Comparator
                .comparingInt(Pair::getValue).reversed() // 首先按照 value 降序排序
                .thenComparing(Pair::getKey)); // 然后按照 key 升序排序

        // 插入一些 Pair 对象
        sortedSet.add(new Pair<>(4, 51));
        sortedSet.add(new Pair<>(8, 85));
        sortedSet.add(new Pair<>(1, 16));
        sortedSet.add(new Pair<>(2, 51));

        // 打印 TreeSet 中的元素
        for (Pair pair : sortedSet) {
            System.out.println("Key: " + pair.getKey() + ", Value: " + pair.getValue());
        }
    }
}

代码解释:

Comparator.comparingInt(Pair::getValue).reversed(): 创建一个 Comparator，首先按照 Pair 对象的 value 值进行比较，并使用 reversed() 方法实现降序排序。
.thenComparing(Pair::getKey): 在 value 值相同的情况下，进一步按照 Pair 对象的 key 值进行升序排序。

输出结果:

Key: 8, Value: 85
Key: 4, Value: 51
Key: 2, Value: 51
Key: 1, Value: 16

可以看到，所有四个 Pair 对

象都被成功插入到 TreeSet 中，并且按照 value 值降序、key 值升序的顺序排列。

注意事项

确保 Pair 类的 getKey() 和 getValue() 方法正确实现，并返回期望的值。
如果需要按照 key 值降序排序，可以使用 .thenComparing(Pair::getKey, Comparator.reverseOrder())。
在更复杂的场景中，可能需要使用更复杂的 Comparator 逻辑来满足特定的排序需求。

总结

通过扩展 Comparator，我们可以有效地解决 TreeSet 中因 Comparator 定义导致具有相同属性值的对象无法插入的问题。关键在于确保 Comparator 能够区分所有不同的对象，即使它们在某些属性上具有相同的值。使用 thenComparing 方法可以方便地实现多级排序，从而满足更复杂的排序需求。