TreeSet

此文为转载：转载出处：点击打开链接

TreeSet是依靠TreeMap来实现的。

TreeSet是一个有序集合，TreeSet中的元素将按照升序排列，缺省是按照自然排序进行排列，意味着TreeSet中的元素要实现Comparable接口。或者有一个自定义的比较器。

我们可以在构造TreeSet对象时，传递实现Comparator接口的比较器对象。

import java.util.Iterator;import java.util.*;public class TreeSetTest {    public static void main(String[] args) {        Set ts = new TreeSet();        ts.add("abc");        ts.add("xyz");        ts.add("rst");        Iterator it = ts.iterator();        while (it.hasNext()) {            System.out.println(it.next());        }    }}

输出：

 

abc

rst

xyz

打印结果不是和先前加入的顺序一样，它是按照一个字母的排序法进行排序的。这是因为String 类实现了Comparable接口。

如果我们自己定义的一个类的对象要加入到TreeSet当中，那么这个类必须要实现Comparable接口。

代码import java.util.Iterator;import java.util.*;public class TreeSetTest {    public static void main(String[] args) {        Set ts = new TreeSet();        ts.add(new Teacher("zhangsan", 1));        ts.add(new Teacher("lisi", 2));        ts.add(new Teacher("wangmazi", 3));        ts.add(new Teacher("mazi", 3));        Iterator it = ts.iterator();        while (it.hasNext()) {            System.out.println(it.next());        }    }}class Teacher implements Comparable {    int num;    String name;    Teacher(String name, int num) {        this.num = num;        this.name = name;    }    public String toString() {        return "学好" + num + "姓名" + name;    }    public int compareTo(Object o) {        Teacher ss = (Teacher) o;        int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);        if (result == 0) {            result = name.compareTo(ss.name);        }        return result;    }}

注意：如果将int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);写成了int result = ss.num > num ? 1 : (ss.num == num ? 0 : -1);那么得到的结果就是倒序排列的，并不是升序的。

 

在使用Arrays对数组中的元素进行排序的时候，可以传递一个比较器。

在使用Collections对集合中的元素进行排序的时候，可以传递一个比较器。

 

那么在使用TreeSet对加入到其中的元素进行排序的时候可以传入一个比较器吗？

那么在使用TreeSet对加入到其中的元素进行排序的时候可以传入一个比较器吗？

 

    public TreeSet(Comparator<? super E> comparator) {
        this(new TreeMap<E,Object>(comparator));
    }

通过查看它的构造方法就知道可以传入一个比较器。

 

构造一个新的空TreeSet，它根据指定比较器进行排序。插入到该 set 的所有元素都必须能够由指定比较器进行相互比较：对于 set 中的任意两个元素 e1 和e2，执行 comparator.compare(e1, e2) 都不得抛出 ClassCastException。如果用户试图将违反此约束的元素添加到 set 中，则 add 调用将抛出 ClassCastException。

代码import java.util.Comparator;import java.util.Iterator;import java.util.TreeSet;public class TreeSetTest {    public static void main(String[] args) {        TreeSet ts = new TreeSet(new Teacher.TeacherCompare());        ts.add(new Teacher("zhangsan", 2));        ts.add(new Teacher("lisi", 1));        ts.add(new Teacher("wangmazi", 3));        ts.add(new Teacher("mazi", 3));        Iterator it = ts.iterator();        while (it.hasNext()) {            System.out.println(it.next());        }    }}class Teacher {    int num;    String name;    Teacher(String name, int num) {        this.num = num;        this.name = name;    }    public String toString() {        return "学号：" + num + "    姓名：" + name;    }    static class TeacherCompare implements Comparator {// 老师自带的一个比较器        public int compare(Object o1, Object o2) {            Teacher s1 = (Teacher) o1;// 转型            Teacher s2 = (Teacher) o2;// 转型            int result = s1.num > s2.num ? 1 : (s1.num == s2.num ? 0 : -1);            if (result == 0) {                result = s1.name.compareTo(s2.name);            }            return result;        }    }}