实现Comparable
package com;
import java.util.*;
public class test {
public static void main(String[] args) {
Person person=new Person(12);
Person person1=new Person(13);
Person person2=new Person(15);
Set set=new TreeSet();
set.add(person);
set.add(person1);
set.add(person2);
for (Object p:set){
System.out.println(p);
}
}
}
class Person implements Comparable<Person>{
private int age;
@Override
public String toString() {
return "Person{" +
"age=" + age +
'}';
}
public Person(int age){
this.age=age;
}
@Override
public int compareTo(Person o) {
return this.age-o.age;
}
}
结果
Person{age=12}
Person{age=13}
Person{age=15}
先按照年龄排序,如果年龄一样,按照姓名排序
package com;
import java.util.*;
/**
* 先按照年龄排序,如果年龄一样,按照姓名排序
*/
public class test {
public static void main(String[] args) {
Set set = new TreeSet();
set.add(new Person(12, "zs"));
set.add(new Person(19, "lsi"));
set.add(new Person(29, "ww"));
for (Object p : set) {
System.out.println(p);
}
}
}
class Person implements Comparable<Person> {
private int age;
private String name;
public Person(int age, String name) {
this.age = age;
this.name = name;
}
@Override
public String toString() {
return "Person{" +
"age=" + age +
", name='" + name + '\'' +
'}';
}
@Override
public int compareTo(Person o) {
if (this.age == o.age) {
return this.name.compareTo(o.name);
} else {
return this.age - o.age;
}
}
}
结果
Person{age=12, name='zs'}
Person{age=19, name='lsi'}
Person{age=29, name='ww'}
使用比较器Comparator
TreeSet集合中元素可排序的第二种方式:使用比较器方式
最终结论:
放到TreeSet或TreeMap集合key部分dd 元素想要做到排序,包括2种方式:
1、放到集合中的元素实现 java.lang.Comparable接口
2、在构造器TreeSet或者TreeMap集合时候给它传一个比较器对象。
Comparator和Comparable怎么选择?
当比较规则不会发生改变时候,或者说当比较规则只有1个时候,建议实现Comparable接口。
如果比较规则有多个,并且需要多个规则之间的频繁转换,建议使用Comparator接口。
Comparator接口符合OCP原则。
package com;
import java.util.*;
public class test {
public static void main(String[] args) {
//匿名内部类方式(这个类没名字,直接new接口)
TreeSet<Person> set = new TreeSet<>(new Comparator<Person>() {
@Override
public int compare(Person o1, Person o2) {
return o1.age - o2.age;
}
});
set.add(new Person(100));
set.add(new Person(200));
set.add(new Person(700));
set.add(new Person(300));
set.add(new Person(900));
for (Object rs : set) {
System.out.println(rs
);
}
}
}
class Person {
int age;
@Override
public String toString() {
return "Person{" +
"age=" + age +
'}';
}
public Person(int age) {
this.age = age;
}
}