本文介绍了在使用 Java `TreeSet` 对 `Pair` 对象进行排序时，如何解决因 `Comparator` 定义导致具有相同 value 值的 `Pair` 对象无法插入的问题。通过扩展 `Comparator`，使得在 value 值相同的情况下，根据 key 值进行进一步排序，从而保证所有不同的 `Pair` 对象都能正确插入到 `TreeSet` 中。
在使用 TreeSet 对自定义对象进行排序时，Comparator 的定义至关重要。如果 Comparator 将两个不同的对象判定为相等，那么 TreeSet 将只会保留其中一个。当我们需要保留所有不同的对象，即使它们在某些属性上具有相同的值时，就需要仔细设计 Comparator。
问题分析
在使用 TreeSet 存储 Pair<Integer, Integer> 对象，并按照 value 值降序排序时，如果两个 Pair 对象的 value 值相同，原始的 Comparator 会认为它们相等。 这导致后续插入具有相同 value 但不同 key 的 Pair 对象时，TreeSet 会拒绝插入，因为它已经包含一个“相等”的对象。
解决方案
为了解决这个问题，我们需要扩展 Comparator，使其在 value 值相同的情况下，进一步比较 key 值。 这样，只有当两个 Pair 对象的 value 和 key 都相同时，才会被认为是相等的。
立即学习“Java免费学习笔记（深入）”；
一种可行的方案是：
- 首先按照 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<Pair<Integer, Integer>> 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<Integer, Integer> 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 方法可以方便地实现多级排序，从而满足更复杂的排序需求。
