1.TreeSet介绍
TreeSet是一个有序集合,可以以任意顺序将元素插入到集合中,在对集合进行遍历的时候,每个元素将自动按照排序后的顺序呈现。底层使用的是二叉树(更具体点是红黑树)实现,对于元素之间排序,如果不指定自定义的比较器Comparator,那么插入的对象必须实现Comparable接口,元素按照实现此接口的compareTo()方法去排序。如果指定了自定义的比较器Comparator,优先使用Comparator去对元素进行排序。比较规则决定了元素是否可以重复,以及元素的排序结果。
2.特点
- 保存的元素按照一定的规则有序
- 可自定义排序规则
- 非线程安全
3.存储Integer类型元素
import java.util.TreeSet;public class DemoTreeSet { public static void main(String[] args) { TreeSetts = new TreeSet<>(); ts.add(1); //自动装箱 ts.add(7); ts.add(3); ts.add(5); ts.add(3); //重复元素 System.out.println(ts); }}输出:[1, 3, 5, 7]
看上面代码的输出,不仅输出元素有序,而且去除了重复元素3,我们在存元素的时候,并不是按照1,3,5,7的顺序存储的,但是遍历却是按照顺序排列的。为什么呢?
前面我们说过,TreeSet在存对象的时候,如果不指定自定义的比较器,那么存储的对象必须实现Comparable接口,好,我们去看下Integer类有没有实现此接口。
public final class Integer extends Number implements Comparable{ //省略其他代码 //...... public int compareTo(Integer anotherInteger) { return compare(this.value, anotherInteger.value); } public static int compare(int x, int y) { return (x < y) ? -1 : ((x == y) ? 0 : 1); }}
分析:
通过Integer源码可知,它满足TreeSet的要求,实现了Comparable接口中唯一的抽象方法compareTo(),在此方法内部又调用了一个compare()方法去定义比较规则,返回值是0,1,或者-1。前面说过,TreeSet底层是二叉树实现的,当存储元素的时候,会调用比较规则方法和二叉树上的元素一一比较,如果要插入的元素比当前元素小就到左子树去比较,如果比当前元素大,就到右子树去比较,直到当前元素的左或者右子树为空,就插入此元素。如果在比较过程中,出现当前元素等于要插入的元素,那么此元素不插入,例如上例中最后一个元素3被过滤掉了,这样也就保证了Set的元素唯一性。
自定义比较器
前面说过,如果我们自定义了比较器Comparator,那么集合会优先使用自定义的比较器去对元素进行排序,好,我们现在想让集合中的Integer元素按照倒序排列,应该如何实现呢?先来查看JDK文档:
上图中红框标注的构造方法就是我们需要的,这里需要传入一个Comparator接口类型的对象,我们当然可以自定义一个类然后实现Comparator接口,然后再new一个对象当作参数传入TreeSet()的构造方法,但是为了方便,我们使用匿名类实现。在此之前,我们先看下Comparator接口都有什么抽象方法。
此接口只有两个抽象方法,我们先不用管equals()方法,定义排序规则需要在compare()这个方法。好,我们来实现一下集合中元素倒序排序。
import java.util.Comparator;import java.util.TreeSet;public class DemoTreeSet { public static void main(String[] args) { TreeSetts = new TreeSet<>(new Comparator () { @Override public int compare(Integer o1, Integer o2) { if(o1 == o2) return 0; return o1 < o2 ? 1 : -1; } }); ts.add(1); //自动装箱 ts.add(7); ts.add(3); ts.add(5); ts.add(3); System.out.println(ts); }}输出:[7, 5, 3, 1]
分析:
1、读者可能发现,Comparator接口明明有两个抽象方法,为什么只实现了compare(),很简单。我们知道匿名类也是一个类,只要是类就会默认继承Object类,那么就会默认继承Object类的equals()方法,所以其实我们已经使用了从Object类继承的方法去实现了Comparator接口的equals()方法。
2、我们在compare()方法中定义了排序规则,和Integer类中实现的方式正好相反,小于返回1,大于返回-1,等于返回0,所以输出的元素就是倒序了。由于我们在TreeSet中自定义了比较器,所以Integer类的比较规则就失效了。
4.存储自定义对象
Student类public class Student implements Comparable{ private String name; private int age; public Student(String name, int age) { super(); this.name = name; this.age = age; } @Override public String toString() { return "Student [name=" + name + ", age=" + age + "]"; } @Override public int compareTo(Student o) { return this.age - o.age; }}
import java.util.TreeSet;public class DemoTreeSet { public static void main(String[] args) { TreeSetts = new TreeSet<>(); ts.add(new Student("张廷玉",43)); ts.add(new Student("纳兰明珠",80)); ts.add(new Student("索额图",79)); System.out.println(ts); }}输出:[Student [name=张廷玉, age=43], Student [name=索额图, age=79], Student [name=纳兰明珠, age=80]]
首先我们定义了一个Student类,前面说过,如果没有自定义比较器,TreeSet集合存储的对象元素必须实现Comparable接口,这里我们的Student类实现了此接口,并且实现了此接口的唯一抽象方法compareTo(),这里我们是按照学生年龄排序(升序)。如果这里不实现Comparable接口,那么会报 java.lang.ClassCastException异常,意思就是这里存的元素不是Comparable类型的。
自定义比较器
我们用自定义比较器来实现一个小案例。这里我们对学生首先按照年龄进行排序,如果年龄相同,我们再按照名字的长度进行排序,都是升序,来看看我们怎么实现它。
public class Student{ private String name; private int age; public Student(String name, int age) { super(); this.name = name; this.age = age; } @Override public String toString() { return "Student [name=" + name + ", age=" + age + "]"; } public String getName() { return name; } public int getAge() { return age; }}
import java.util.Comparator;import java.util.TreeSet;public class DemoTreeSet { public static void main(String[] args) { TreeSetts = new TreeSet<>(new Comparator () { @Override public int compare(Student o1, Student o2) { int num = o1.getAge() - o2.getAge(); return num == 0 ? o1.getName().length() - o2.getName().length() : num; } }); ts.add(new Student("张廷玉",43)); ts.add(new Student("纳兰明珠",80)); ts.add(new Student("索额图",79)); ts.add(new Student("苏麻喇姑",79)); for (Student student : ts) { System.out.println(student); } }}输出:Student [name=张廷玉, age=43]Student [name=索额图, age=79]Student [name=苏麻喇姑, age=79]Student [name=纳兰明珠, age=80]
由于我们在TreeSet中自定义了比较器,这里的Student类不需要实现Comparable接口了。我们来看compare()方法内部,首先计算比较的元素的年龄差num,如果num不等于0,那么我们还是按照年龄排序,如果年龄相同,那么我们返回元素名字长度的差,也就是说年龄相同,我们按照名字长度进行排序。从输出可以看出,索额图和苏麻喇姑的年龄相同,但是索额图名字长度要小于苏麻喇姑,所以索额图排在了苏麻喇姑的前面输出。
至此!如何使用TreeSet去存储对象以及内部的排序原理我们就讲完了,有时间自己动手去实现一下,多思考为什么要这样,眼高手低是程序员一大忌讳。如果不懂的读者可以在下面留言!