只要放在TreeSet中的元素对象,在该对象的类中必须实现Comparable接口,必须覆盖该接口中的compareTo()方法,并在该方法中编写比较规则。(该方法不能自动生成)
- 自然排序(Comparable)
TreeSet拥有的集合元素的compareTo()方法来比较元素的大小关系,然后将集合元素按照升序排列。
int compareTo(T o)
将此对象与指定的对象进行比较,以返回一个负整数、零或一个正整数,因为这个对象小于、等于或大于指定的对象。参数 : o要比较的对象。
结果 : 一个负整数,零,或一个正整数,因为这个对象小于,等于,或大于指定的对象。
import java.util.HashSet;
import java.util.Set;
import java.util.TreeSet;
public class HashSetDemo {
public static void main(String[] args) {
TreeSet<Student> set=new TreeSet<>();
set.add(new Student(1,"Tom",17));
set.add(new Student(2,"Will",22));
set.add(new Student(1,"Tom",19));
set.add(new Student(1,"Jery",18));
//对象的年龄各不相同,所以认为是四个不同的对象
System.out.println(set.size());//4
System.out.println(set);
/* [Student [id=1, name=Tom, age=17],
Student [id=1, name=Jery, age=18],
Student [id=1, name=Tom, age=19],
Student [id=2, name=Will, age=22]]*/
}
}
class Student implements Comparable<Student>{//泛型
int id;
String name;
int age;
public Student(int id, String name, int age) {
this.id = id;
this.name = name;
this.age = age;
}
//-----------------------------------------------
@Override
public int compareTo(Student o) {//默认的自然排序
//按照年龄的大小排序,同时去重。
if(this.age>o.age){
return 1;
}
if(this.age<o.age){
return -1;
}
return 0;
/*
//由于是自定义对象,需要自己写ComparaTo()也可以和下面定制排序一样自定义排序规则.
if(this.name.length()>o.name.length()){
return -1;
}
if(this.name.length()<o.name.length()){
return 1;
}
return 0;
*/
}
//-----------------------------------------------
@Override
public String toString() {
return "Student [id=" + id + ", name=" + name + ", age=" + age + "]";
}
}
- 定制排序(Comparator)
第三方比较器,比较时需要创建第三方类,实现Comparator接口,并且覆盖其中的compare()方法,编写比较规则和排序方式。
int compare(T o1, T o2)
比较其两个顺序的参数。返回一个负整数、零或一个正整数作为第一个参数小于、等于或大于第二个参数。
import java.util.Comparator;
import java.util.HashSet;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo {
public static void main(String[] args) {
//使用定制排序需要给TreeSet对象传递一个第三方比较器的对象
TreeSet<Student> set=new TreeSet<>(new NameLengthSort());
set.add(new Student(1,"Tm",17));
set.add(new Student(2,"Will",22));
set.add(new Student(1,"Tom",19));
set.add(new Student(1,"Jerry",18));
//对象name的长度各不相同,所以认为是四个不同的对象
System.out.println(set.size());//4
System.out.println(set);
/* [Student [id=1, name=Jerry, age=18],
* Student [id=2, name=Will, age=22],
* Student [id=1, name=Tom, age=19],
* Student [id=1, name=Tm, age=17]]*/
}
}
class Student{
int id;
String name;
int age;
public Student(int id, String name, int age) {
this.id = id;
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Student [id=" + id + ", name=" + name + ", age=" + age + "]";
}
}
//按照名字的长度进行排序
class NameLengthSort implements Comparator<Student>{
@Override
public int compare(Student o1, Student o2) {
if(o1.name.length()>o2.name.length()){
return -1;
}
if(o1.name.length()<o2.name.length()){
return 1;
}
return 0;
}
}
- 自然排序和定制排序的比较
- 总结:
1.当向TreeSet里面添加不是自定义对象时,元素会默认使用自然排序输出从小到大序列。
2.当向TreeSet里面添加自定义的对象时,可以使用这2种对象比较的方法没有区别。也可以在compareTo()方法中自定义比较规则和排序规则,个人认为使用实现Comparator会更方便。