从头认识java-15.4 使用TreeSet需要注意的地方

时间:2021-05-02 06:52:42

接着上一章节,我们继续来讨论使用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需要注意的地方。


这一章节就到这里,谢谢。

-----------------------------------

目录