谈一个比较基础,又很常用的东西, Comparator 类,之前我写过一篇浅浅的关于这个的文章 ,今天再复盘一下这个问题,把它弄熟
ps: 本文中提供的代码,为了提高可读性都没用 lamdam 和 函数式编程 简化书写,大家平时用的时候,自己使用 IDEA 快捷键转化一下
一、 做什么
Comparator 类常作为 sorted() 方法的参数传递给 sorted 方法,用来解决给集合排序,自定义排序规则的问题 。
那从这个角度看,这个类肯定很常用了,因为我知道大家都不喜欢自己写排序方法,而且自己写的肯定还有性能优化问题,我们直接用 jdk 帮我们封装好的,就即简洁,又易于阅读 。
二、使用
1. 对 Array 数组类型进行排序
下面是 Arrays 类暴露给我们的排序方法 —— sorted
Stream<T> sorted(Comparator<? super T> comparator);
Comparator 接口暴露给我们以下两个方法给我们实现:
@FunctionalInterface
int compare(T o1, T o2);
boolean equals(Object obj);
还有以下几个静态方法,给我们直接用类名调用
(1)情况一 : 直接 new Comparator 对象,然后实现其 compare 方法进行排序
这种情况可以在 compare 内部定义好排序规则,看起来非常直观,通过在 compare 中写好逻辑代码,可以实现多种排序规则 。
Integer[] a = {1, 2, 3, 0};
// 对 a 按照逆序排序
Integer[] a2 = Arrays.stream(a).sorted(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
}).toArray(new IntFunction<Integer[]>() {
@Override
public Integer[] apply(int value) {
return new Integer[value];
}
});
(2)情况二: 使用 Comparator 的静态方法,链式的添加排序规则
这种方式也可以随意定制排序规则,而且相对来说,代码的可读性更高,如下实例:
Integer[] a1 = Arrays.stream(a).sorted(Comparator.comparing(new Function<Integer, Integer>() {
@Override
public Integer apply(Integer integer) {
return integer;
}
}).reversed()).toArray(new IntFunction<Integer[]>() {
@Override
public Integer[] apply(int value) {
return new Integer[value];
}
});
其提供了如下方法,供我们使用 : 其中 then 那一堆可以帮我们定义二级排序规则,也就是说第一级相等的时候,按照那个属性进行排序
2. 对 Collection 集合类型进行排序
总体思想和上面的差不多,只不过排序方法变成了 sort
下面是一个按照 People 类的身高进行排序的例子 :
class Solution {
class People {
public String name;
public int height;
public People(String name, int height){
this.name = name;
this.height = height;
}
public int getHeight(){
return height;
}
}
public String[] sortPeople(String[] names, int[] heights) {
List<People> list = new ArrayList<>(getPeopleList(names, heights));
Collections.sort(list, Comparator.comparing(People :: getHeight).reversed());
String[] ans = list.stream().map(people -> people.name).toArray(String[]::new);
return ans;
}
public List<People> getPeopleList(String[] names, int[] heights){
List<People> list = new ArrayList<>();
for(int i=0; i<names.length; i++){
People people = new People(names[i], heights[i]);
list.add(people);
}
return list;
}
}
三、总结
这文章写的乱七八糟的,抱歉大家,我最近脑子比较乱,所以不总结都不行了,总结就以下两点:
- 直接创建 Comparator 类实例, 然后重写 compare 方法定义排序规则
- 使用 Comparator 静态方法进行排序,直接用类名调用进行排序,给方法传入相应的排序指标