接着上一章节,我们继续来讨论使用TreeSet需要注意的地方。
2.TreeSet
特点:
(1)元素是有序的
(2)元素是重复的,需要去重
(3)必须实现Comparable接口
(4)底层的数据存储是按照树的结构来存储的
我们修改一下上一章节的代码:
package com.ray.ch15;
import java.lang.reflect.InvocationTargetException;
import java.util.Set;
import java.util.TreeSet;
public class Test<T> {
public static <T> Set<T> fill(Set<T> set, Class<T> type)
throws InstantiationException, IllegalAccessException,
IllegalArgumentException, SecurityException,
InvocationTargetException, NoSuchMethodException {
for (int i = 0; i < 10; i++) {
set.add(type.getConstructor(int.class).newInstance(i));
}
return set;
}
public static <T> void test(Set<T> set, Class<T> type)
throws IllegalArgumentException, SecurityException,
InstantiationException, IllegalAccessException,
InvocationTargetException, NoSuchMethodException {
fill(set, type);
fill(set, type);
fill(set, type);
System.out.println(set);
}
public static void main(String[] args) throws IllegalArgumentException,
SecurityException, InstantiationException, IllegalAccessException,
InvocationTargetException, NoSuchMethodException {
test(new TreeSet<TreeType>(), TreeType.class);
// test(new TreeSet<SetType>(), SetType.class);//
// java.lang.ClassCastException
// test(new TreeSet<HashType>(), HashType.class);//
// java.lang.ClassCastException
}
}
class SetType {
private int id = 0;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public SetType(int i) {
id = i;
}
@Override
public String toString() {
return id + "";
}
@Override
public boolean equals(Object obj) {
return obj instanceof SetType && (id == ((SetType) obj).id);
}
}
class HashType extends SetType {
public HashType(int i) {
super(i);
}
@Override
public int hashCode() {
return getId();
}
}
class TreeType extends HashType implements Comparable<TreeType> {
public TreeType(int i) {
super(i);
}
@Override
public int compareTo(TreeType o) {
if (o.getId() > getId()) {// 排序
return -1;
} else {
if (o.getId() == getId()) {// 去重
return 0;
} else {
return 1;
}
}
}
}
输出:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
解释一下上面的代码:
(1)fill方法:通过泛型和反射,往传进来的set里面添加数据
(2)test方法:我们通过多次的往set里面填充数据,看看set是否去重
(3)SetType:原始类型,只是简单实现了equals方法和toString方法,toString这里通过输出id来表示对象
(4)HashType:继承SetType,再实现了hashCode方法抛异常
(5)TreeType:实现Comparable接口,我们在compareTo方法里面已经做了去重和排序,如果只是做其中一个,只是实现一种功能。
总结:这一章节主要讨论了使用TreeSet需要注意的地方。
这一章节就到这里,谢谢。
-----------------------------------