Java之TreeSet介绍

import java.util.Comparator;import java.util.TreeSet;import org.junit.Test;//非线程安全public class TreeSetTest {/** * TreeSet:它可以给Set集合中的元素进行指定方式的排序。 保证元素唯一性的方式：通过比较的结果是否为0. 底层数据结构是：二叉树。 *  * 排序的第一种方式： 让元素自身具备比较性。只要让元素实现Comparable接口，覆盖compareTo方法即可。 *  * 但是，如果元素自身不具备比较性，或者元素自身具备的比较性，不是所需要的。 * 比如，学生的自然排序是按年龄排序，现在想要按照学生姓名排序。还可以不改动原有代码。 这时怎么办呢？ *  * 排序的第二种方式：自定比较器的方式。这时可以让集合自身具备比较性。 *  * 可以定义一个类实现Comparator接口，覆盖compare方法。 * 将该Comparator接口子类对象作为实际参数传递给TreeSet集合构造函数。 */public static void main(String[] args) {TreeSet<Student> ts = new TreeSet<Student>();// Student如果不实现Comparable，add时会出错，Student cannot be cast to java.lang.Comparable//添加第一个对象时，TreeSet里没有任何元素，所以没有问题；//当添加第二个Err对象时，TreeSet就会调用该对象的compareTo(Object obj)方法与集合中其他元素进行比较——//如果对应的类没有实现Comparable接口，则会引发ClassCastException异常// add方法内部会对插入的数据进行排序，此时元素自身具备比较性，因为其实现了Comparable接口ts.add(new Student("lisi0", 30));ts.add(new Student("lisixx", 29));ts.add(new Student("lisi9", 29));ts.add(new Student("lisi8", 38));// 重复插入无效，但是不报错（根据年龄和名字进行比较，都相同，视为同一个学生）ts.add(new Student("lisixx", 29));ts.add(new Student("lisixx", 28));ts.add(new Student("lisi4", 14));ts.add(new Student("lisi7", 27));System.out.println(ts);}//此时根据自定义的比较器进行排序@Testpublic void testComparator() {// 可以利用匿名类，按名字进行排序TreeSet<Student> ts = new TreeSet<Student>(new Comparator<Student>() {@Overridepublic int compare(Student s1, Student s2) {int num = s1.getName().compareTo(s2.getName());return num;}});ts.add(new Student("lisi0", 30));ts.add(new Student("lisixx", 29));ts.add(new Student("lisi9", 29));ts.add(new Student("lisi8", 38));//此时根据名字进行排序，不管年龄是否相同，下面两个都不会插入//名字相同，则视为同一个人ts.add(new Student("lisixx", 29));ts.add(new Student("lisixx", 28));ts.add(new Student("lisi4", 14));ts.add(new Student("lisi7", 27));System.out.println(ts);}//奇数升序，偶数降序@Testpublic void mySort() {TreeSet set = new TreeSet(new Comparator() {int r = 0;@Overridepublic int compare(Object a, Object b) {int n1 = Integer.parseInt(a.toString());int n2 = Integer.parseInt(b.toString());if (n1 % 2 != n2 % 2) {r = (n2 % 2 - n1 % 2);} else if (n1 % 2 == 1) {if (n1 > n2)r = 1;else if (n1 < n2)r = -1;} else if (n1 % 2 == 0) {if (n1 > n2)r = -1;else if (n1 < n2)r = 1;}return r;}});set.add("1");set.add("2");set.add("3");set.add("4");set.add("5");set.add("6");set.add("7");set.add("8");set.add("9");set.add("10");System.out.println(set);}// //同姓名同年龄的学生视为同一个学生public static class Student implements Comparable<Student> {private int age;private String name;public Student(String name, int age) {this.age = age;this.name = name;}@Overridepublic int compareTo(Student stu) {int num = new Integer(this.age).compareTo(new Integer(stu.age));return num == 0 ? this.name.compareTo(stu.name) : num;}public String toString() {return name + "::" + age;}public int getAge() {return age;}public void setAge(int age) {this.age = age;}public String getName() {return name;}public void setName(String name) {this.name = name;}}}