TreeSet是Java Collection Framework的一个通用类，它实现了SortedSet接口。它将集合的元素存储在树结构中。此外，所有的元素都以排序的方式存储，并且如果我们尝试添加自定义类对象，它们必须是可相互比较的，否则我们将遇到java.lang.ClassCastException异常。这里，自定义类对象指的是使用构造函数创建的用户定义对象。修复这个ClassCastException的一种方法是使用自定义比较器。让我们详细讨论一下
TreeSet 的一般语法如下：
TreeSet<TypeOfSet> nameOfSet = new TreeSet<>();
在本节中，我们将解释如何使用自定义比较器来解决TreeSet中的ClassCastException问题。让我们从比较器开始讨论
正如其名称所示，它用于比较某些内容。在Java中，Comparator是一个接口，用于对自定义对象进行排序。我们可以在其内置的名为'compare()'的方法中编写自己的逻辑来对指定的对象进行排序。此方法接受两个对象作为参数，然后返回一个整数值。通过这个整数值，Comparator决定哪个对象更大
class nameOfComparator implements Comparator< TypeOfComparator >() {
compare( type object1, type object2 ) {
// logic for comparison
}
}
在这里，class是创建一个类的关键字，implements是启用接口提供的功能的关键字。
在下面的示例中，我们将尝试向TreeSet添加自定义类对象，而不使用Comparator来展示Java编译器抛出java.lang.ClassCastException的情况。
import java.util.*;
public class ShopBag {
String item;
int price;
ShopBag(String item, int price) { // constructor
// this keyword shows these variables belong to constructor
this.item = item;
this.price = price;
}
public static void main(String[] args) {
// Declaring collection TreeSet
TreeSet<ShopBag> trSet = new TreeSet<ShopBag>();
// Adding object to the collection
trSet.add(new ShopBag("Rice", 59));
trSet.add(new ShopBag("Milk", 60));
// to print the objects
for (ShopBag print : trSet) {
System.out.println("Item: " + print.item + ", " + "Price: " + print.price);
}
}
}
Exception in thread "main" java.lang.ClassCastException: class ShopBag cannot be cast to class java.lang.Comparable (ShopBag is in unnamed module of loader 'app'; java.lang.Comparable is in module java.base of loader 'bootstrap') at java.base/java.util.TreeMap.compare(TreeMap.java:1569) at java.base/java.util.TreeMap.addEntryToEmptyMap(TreeMap.java:776) at java.base/java.util.TreeMap.put(TreeMap.java:785) at java.base/java.util.TreeMap.put(TreeMap.java:534) at java.base/java.util.TreeSet.add(TreeSet.java:255) at ShopBag.main(ShopBag.java:14)
以下示例说明了在修复ClassCastException时使用Comparator的方法。
首先，导入 'java.util' 包，以便我们可以使用 TreeSet
创建一个名为 'ShopBag' 的类。在类内部，声明两个变量，并定义一个构造函数，该构造函数带有两个参数 'item' 和 'price'，分别为字符串和整数类型。
然后，定义另一个实现Comparator接口的类'Comp'，在其中使用'compare()'方法对TreeSet进行升序排序。
在'main()'方法中，通过传递'Comp'类的实例来创建一个TreeSet集合，以便它可以进行排序
最后，使用'add()'方法将一些元素存储到TreeSet集合中，然后使用for-each循环打印集合的元素。
import java.util.*;
class ShopBag {
String item;
int price;
ShopBag(String item, int price) { // constructor
// this keyword shows these variables belong to constructor
this.item = item;
this.price = price;
}
}
// use of comparator interface
class Comp implements Comparator<ShopBag> {
// logic to sort
public int compare(ShopBag i, ShopBag j) {
if(i.price > j.price) {
return 1;
} else {
return -1;
}
}
}
public class Example2 {
public static void main(String[] args) {
// Declaring collection TreeSet
TreeSet<ShopBag> trSet = new TreeSet<ShopBag>(new Comp());
// Adding object to the collection
trSet.add(new ShopBag("Rice", 59));
trSet.add(new ShopBag("Milk", 60));
trSet.add(new ShopBag("Bread", 45));
trSet.add(new ShopBag("Peanut", 230));
trSet.add(new ShopBag("Butter", 55));
System.out.println("Objects of the collection: ");
// to print the objects
for (ShopBag print : trSet) {
System.out.println("Item: " + print.item + ", " + "Price: " + print.price);
}
}
}
Objects of the collection: Item: Bread, Price: 45 Item: Butter, Price: 55 Item: Rice, Price: 59 Item: Milk, Price: 60 Item: Peanut, Price: 230
我们通过定义TreeSet并介绍TreeSet的ClassCastException来开始这篇文章。在下一节中，我们介绍了Comparator接口，它可以帮助解决ClassCastException。然后，我们讨论了两个示例程序，展示了ClassCastException的情况以及如何修复这个异常。
