TreeSet想要实现自定义类的自动排序有两种方式:一种是实现Comparable<>接口,另一种是比较器的方法(也就是给这个类创建一个比较器进行比较)
这个文章主要说明比较器的方式
创建一个比较器有两种不同的方式,一种是创建一个比较器类实现comparator<>接口,另一种是在创建集合时直接new出comparator ,
创建比较器的第一种方式:需求就是先按照age从小到大排序,如果age相同就按照name的进行排序
public class Map08 { public static void main(String[] args) { // 如果要实现自动排序TreeMap或treeset是必不可少的 // 本例子采用set集合进行演示 Set<Customers> set = new TreeSet<>(new CCompartor()); set.add(new Customers("zhangsi",20)); set.add(new Customers("zhangsan",20)); set.add(new Customers("mike",10)); set.add(new Customers("join",17)); for(Customers customers : set){ System.out.println(customers); } } }
这是自己定义的类,就是要把这个类里面的数据进行自动排序
class Customers{ String name; int age; public Customers(String name, int age) { this.name = name; this.age = age; } @Override public String toString() { return "Customers{" + "name='" + name + '\'' + ", age=" + age + '}'; } } //写出自定义类的比较器,需要实现compatator接口,同时重写里面的方法 //使用比较器其实还有另外一个方法 class CCompartor implements Comparator<Customers> { @Override public int compare(Customers o1, Customers o2) { if(o1.age == o2.age){ return o1.name.compareTo(o2.name); }else{ return o1.age - o2.age; } } }
输出结果如下:
Customers{name='mike', age=10}
Customers{name='join', age=17}
Customers{name='zhangsan', age=20}
Customers{name='zhangsi', age=20}
接下来是创建比较器的第二种方式:
public class Map08 { public static void main(String[] args) {
Set<Customers> set1 = new TreeSet<>(new Comparator<Customers>() { @Override public int compare(Customers o1, Customers o2) { if(o1.age == o2.age){ return o1.name.compareTo(o2.name); }else{ return o1.age - o2.age; } } }); set1.add(new Customers("zhangsi",20)); set1.add(new Customers("zhangsan",20)); set1.add(new Customers("mike",10)); set1.add(new Customers("join",17)); for(Customers c : set1){ System.out.println(c); } } }
class Customers{ String name; int age; public Customers(String name, int age) { this.name = name; this.age = age; } @Override public String toString() { return "Customers{" + "name='" + name + '\'' + ", age=" + age + '}'; } }
这种方式就是直接在创建集合的时候进行比较器的创建,就不用再自己创建一个比较器的类了
创建比较器和第一种方式相比比较灵活,如果需要时常更改排序规则,就使用比较器的方法,
如果排序的规则不需要更改或者更改较少的话,就使用comparable<>接口的方法
以上就是运用比较器的方法实现TreeSet集合的自动排序的代码。。。