TreeSet集合概述和特点
TreeSet集合特点
- 元素有序: 这里的顺序不是指存储和取出的顺序,而是按照一定的规则进行排序,具体排序方式取决于构造方法;
TreeSet():根据其元素的自然排序进行排序
TreeSet(Comparator comparator) :根据指定的比较器进行排序
- 没有带索引的方法, 所以不能使用普通for循环遍历,可使用迭代器和增强for循环;
- 由于是Set集合,所以不包含重复元素的集合;
TreeSet集合自排序的方式:
- 有值特性的元素直接可以升序排序。(浮点型, 整型)
- 字符串类型的元素会按照首字符的ASCII排序。
- 对于自定义的引用数据类型,Treeset默认无法排序,执行的时候直接报错,因为人家不知道排序规则。
存储整数并遍历:
public class TreeSetTest {
public static void main(String[] args) {
TreeSet<Integer> ts = new TreeSet<>();
ts.add(10);
ts.add(40);
ts.add(20);
ts.add(98);
ts.add(75);
ts.add(40);
for (Integer i : ts) {
System.out.println(i);
}
/*
* 结果: 10 20 40 75 98
*/
}
}
自定义的引用数据类型的排序实现:
对于自定义的引用数据类型,TreeSet默认无法排序
所以我们需要定制排序的大小规则,程序员定义大小规则的方案有2种:
- 直接为对象的类实现比较器规则接口Comparable,重写compareTo比较方法
Java规则:
如果程序员认为比较者大于被比较者,返回正数
如果程序员认为比较者小于被比较者,返回负数
如果程序员认为比较者等于被比较者,返回0 - 直接为集合设置Comparator对象,重写比较方法
java规则还是一样
实际开发在类上重写,或为集合设置比较器。
如果两种比较规则都存在,则会优先使用选择集合设置的比较器。
1. 自然排序Comparable的使用
存储学生对象并遍历,创建TreeSet集合使用无参构造方法
- 要求: 按照年龄从小到大排序,年龄相同时,按照姓名的字母顺序排序
public class Student implements Comparable<Student> {//实现接口
private String name;
private int age;
public Student() {
}
public Student(String name, int age) {
super();
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
// 重写compareTo
@Override
public int compareTo(Student s) {
// return 0; //返回0则认为是同一个元素,不添加
// return 1; //返回正数是升序
// return -1; //返回负数是倒序
// 按照年龄从小到达排序
return (this.age - s.age) == 0 ;
}
}
public class TreeSetTest {
public static void main(String[] args) {
TreeSet<Student> ts = new TreeSet<>();
Student s1 = new Student("a小明", 10);
Student s2 = new Student("b小红", 21);
Student s3 = new Student("c小李", 22);
Student s4 = new Student("d小蓝", 10);
Student s5 = new Student("d小蓝", 10);
ts.add(s1);
ts.add(s2);
ts.add(s3);
ts.add(s4);
for (Student s : ts) {
System.out.println(s.getName() + "," + s.getAge());
}
/*
* a小明,10 d小蓝,10 b小红,21 c小李,22
*/
}
}
结论
- 用TreeSet集合存储自定义对象, 无参构造方法使用的是自然排序对元素进行排序的;
- 自然排序, 就是让元素所属的类实现Comparable接口,重写compareTo(T o)方法;
- 重写方法时,一定要注意排序规则必须按照要求的主要条件和次要条件来写;
2.比较器排序Comparator的使用
- 存储学生对象并遍历,创建TreeSet集合使用带参构造方法
- 要求: 按照年龄从小到大排序,年龄相同时,按照姓名的字母顺序排序
public class Student {
private String name;
private int age;
public Student() {
}
public Student(String name, int age) {
super();
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
};
}
public class TreeSetTest {
public static void main(String[] args) {
TreeSet<Student> ts = new TreeSet<>(new Comparator<Student>() {
@Override
public int compare(Student s1, Student s2) {
// this.age - s.age
// s1,s2
return s1.getAge() - s2.getAge();
}
});
Student s1 = new Student("a小明", 10);
Student s2 = new Student("b小红", 21);
Student s3 = new Student("c小李", 22);
Student s4 = new Student("d小蓝", 10);
Student s5 = new Student("d小蓝", 10);
ts.add(s1);
ts.add(s2);
ts.add(s3);
ts.add(s4);
for (Student s : ts) {
System.out.println(s.getName() + "," + s.getAge());
}
}
}
结论
- 用TreeSet集合存储自定义对象, 带参构造方法使用的是比较器排序对元素进行排序的;
- 比较器排序,就是让集合构造方法接收Comparator的实现类对象,重写compare(T o1,T o2)方法;
- 重写方法时, 一定要注意排序规则必须按照要求的主要条件和次要条件来写;