Java 中TreeSet 是Set的一个子类。
Set、List、Map区别
Set是一个无序、不允许重复的集合。
List(ArrayList、Vector等)是有序、可重复的。
Map(HashMap等)是键值对。
public static void demo() {
TreeSet<Person> ts = new TreeSet<>();
ts.add(new Person("张三", 23));
ts.add(new Person("李四", 13));
ts.add(new Person("周七", 13));
ts.add(new Person("王五", 43));
ts.add(new Person("赵六", 33));
System.out.println(ts);
}
此处会报异常,异常类型为java.lang.ClassCastException
这是因为Person类没有实现Comparable 接口,并实现compareTo函数。
compareTo函数就是TreeSet用来判断是否唯一的函数。
public class TreeEntity implements Comparable<Object> {
@JsonProperty(value = "Name")
private String name = "";
/**
* 名称(字段名)
*
*/
public final String getName() {
return name;
}
public final void setName(String value) {
name = value;
}
@JsonProperty(value = "Header")
private String header = "";
/**
* 显示的名称(字段别名)
*/
public final String getHeader() {
if (header.isEmpty()) {
header = name;
}
return header;
}
public final void setHeader(String value) {
header = value;
}
@Setter(AccessLevel.PROTECTED)
@JsonProperty(value = "Childrens")
private TreeSet<TreeEntity> childrens;
/**
* 子节点集合
*
*/
public final TreeSet<TreeEntity> getChildrens() {
return childrens;
}
public final void setChildrens(TreeSet<TreeEntity> value) {
childrens = value;
}
@Override
public int compareTo(Object o) {
TreeEntity te = (TreeEntity) o;
if (te == null)
return 1;
return this.getName().compareTo(te.getName());
}
/**
* 构造函数
*/
public TreeEntity() {
childrens = new TreeSet<TreeEntity>();
}
}
}
}
入上图中的TreeEntity类重写了compareTo函数,则是通过name属性来判断是否唯一。
在TreeSet.add()函数中,会触发此compareTo函数,如果判断不唯一,不会添加进去,但是代码不会报异常。
compareTo函数返回值说明:
为什么返回0,只会存一个元素,返回-1会倒序存储,返回1会怎么存就怎么取呢?原因在于TreeSet底层其实是一个二叉树机构,且每插入一个新元素(第一个除外)都会调用compareTo()
方法去和上一个插入的元素作比较,并按二叉树的结构进行排列。
1. 如果将compareTo()
返回值写死为0,元素值每次比较,都认为是相同的元素,这时就不再向TreeSet中插入除第一个外的新元素。所以TreeSet中就只存在插入的第一个元素。
2. 如果将compareTo()
返回值写死为1,元素值每次比较,都认为新插入的元素比上一个元素大,于是二叉树存储时,会存在根的右侧,读取时就是正序排列的。
3. 如果将compareTo()
返回值写死为-1,元素值每次比较,都认为新插入的元素比上一个元素小,于是二叉树存储时,会存在根的左侧,读取时就是倒序序排列的。