Java Set集合之Treeset

TreeSet它可以给Set集合中的元素进行指定方式的排序。

保证元素唯一性的方式：通过比较的结果是否为0.

底层数据结构是：二叉树。

排序的第一种方式：

让元素自身具备比较性。只要让元素实现Comparable接口，覆盖compareTo方法即可。

但是，如果元素自身不具备比较性，或者元素自身具备的比较性，不是所需要的。

比如，人的自然排序是按年龄排序，现在想要按照人的姓名排序。还可以不改动原有代码，此时可用第二种排列方式

排序的第二种方式：自定比较器的方式。
这时可以让集合自身具备比较性。

可以定义一个类实现Comparator接口，覆盖compare方法。将该Comparator接口子类对象作为实际参数传递给TreeSet集合构造函数。该对象就是比较器。

以下例子使用到了泛型，

第一种排列方式例子：People类实现的comparable接口具备了比较性，具体的排序规则是通过compareTo方法实现的

/**
* Comparable接口,使得People本身具备比较性,具体如何排序通过重写{@link People#compareTo(People)} 实现排序
*
* @author TOO-YOUNG-TOO-SIMPLE
*/
class People implements Comparable<People> {
People(String name, int age) {
this.name = name;
this.age = age;
}

private String name;
private int age;

public String getName() {
return name;
}

public int getAge() {
return age;
}

/**
* @return 0:表示重复元素
*/
@Override
public int compareTo(People o) {
if (this.age == o.age) {// 年龄为排序首要条件,年龄相同下判断名字
return this.name.compareTo(o.name);
}
return this.age - o.age;
}

@Override
public String toString() {
return "People [name=" + name + ", age=" + age + "]";
}
}

模拟添加数据：

public static void main(String[] args) {
TreeSet<People> people = new TreeSet<People>();
people.add(new People("tree3", 11));
people.add(new People("tree1", 12));
people.add(new People("tree2", 11));
people.add(new People("tree2", 15));
Iterator<People> p = people.iterator();
while (p.hasNext()) {
System.out.println(p.next().toString());
}

}

控制台结果：由于排序条件为：年龄小的优先，年龄一样再按照名称排序最终结果如下：



第二种排列方式例子：People类不管是否实现comparable接，即不管本身是否具备比较性，只要实例化TreeSet时实现Comparator接口，并重写compare方法，则最终排序将以compare方法中的规则排序，换句话说Comparator的优先级别高于Comparable，此时如果元素(People)实现了Comparable接口，其抽象方法compareTo将被覆盖.

/**
* 实现Comparator接口重写{@link MyComp#compare(People, People)}方法实现排序规则
*
* @author Administrator
*
*/
class MyComp implements Comparator<People> {

/**
* @return 0：表示元素存在，大于0表示按插入顺序排序，小于0表示插入到上个元素前
*/
@Override
public int compare(People o1, People o2) {
if (o1.getName().compareTo(o2.getName()) == 0) {// 首先判断名字是否相同，相同再判断年龄是否相同
return o1.getAge() - o2.getAge();
}
return o1.getName().compareTo(o2.getName());
}
}

模拟数据：构造TreeSet时传入实现了Comparator的类，此时People类中的compareTo方法已经不起作用，上面的代码是新的插入排序规则：



最终输出的结果如下：按照名字优先进行排序，名字相同再按照年龄排序