目录
最近在做算法题的时候,发现排序在大部分题中都不可或缺,今天心血来潮,总结下Java中集合排序常用的方法,基本覆盖了大部分的排序场景(特别是自定义比较器,太好用叻( ̄▽ ̄)/)
文中提到的自然顺序排序,对于可比较大小的数据类型,一般按照从小到大的顺序,无法比较大小的,一般指按字典序升序排列
Array · 数组
使用Arrays.sort()方法对数组进行排序
int[] arr = {1, 5, 2, 1, 4};
System.out.println("排序前:"+Arrays.toString(arr));
Arrays.sort(arr);
System.out.println("排序后:"+Arrays.toString(arr));
List · 列表
Collections.sort()
简单类型
使用Collections.sort()方法对List进行排序,可以对List进行原地排序(也就是不创建新的List对象),如下所示
List<String> list = new ArrayList<>();
list.add("b");
list.add("d");
list.add("ca");
list.add("da");
System.out.println("排序前:"+list);
Collections.sort(list);
System.out.println("排序后:"+list);
复杂对象
当List中的对象比较复杂,当默认的排序无法满足需求的时候,可以使用Comparator接口,Comparator可以进行自定义的比较逻辑来控制排序方式
List<Person> list = new ArrayList<>();
list.add(new Person("李四1", 19));
list.add(new Person("李四2", 29));
list.add(new Person("李四3", 39));
list.add(new Person("李四4", 49));
list.add(new Person("张三1", 59));
list.add(new Person("张三2", 69));
list.add(new Person("张三3", 79));
list.add(new Person("张三4", 89));
System.out.println("按名字排序前:" + list);
Collections.sort(list, new Comparator<Person>() {
@Override
public int compare(Person o1, Person o2) {
//按名字排序
return o1.getName().compareTo(o2.getName());
}
});
System.out.println("按名字排序后:" + list);
System.out.println("按年龄排序前:" + list);
Collections.sort(list, new Comparator<Person>() {
@Override
public int compare(Person p1, Person p2) {
return p1.getAge() - p2.getAge();
}
});
System.out.println("按年龄排序后:" + list);
类
可以使用Comparable接口进行自然排序 如果有一个类需要排序,可以让改类实现Comparable接口,并实现其compareTo()方法,这样,当要对这个类的对象进行排序时,就可以直接使用Collections.sort()方法
List<Person> list = new ArrayList<>();
list.add(new Person("李四1", 19));
list.add(new Person("李四2", 29));
list.add(new Person("李四3", 39));
list.add(new Person("李四4", 49));
list.add(new Person("张三1", 59));
list.add(new Person("张三2", 69));
list.add(new Person("张三3", 79));
list.add(new Person("张三4", 89));
System.out.println("按年龄排序前:" + list);
Collections.sort(list);
System.out.println("按年龄排序后:" + list);
static class Person implements Comparable<Person> {
private String name;
private int age;
public Person(String name, int age) {
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public int getAge() {
return age;
}
@Override
public String toString() {
return "[" + name + "," + age + "]";
}
@Override
public int compareTo(Person o) {
return Integer.compare(this.getAge(), o.getAge());
}
}
使用Lambda表达式
list.sort((o1, o2) -> o1.compareTo(o2));
Stream API
使用Stream API排序,Java 8还引入了Stream API,它提供了一种流式操作的方式来处理集合数据,Stream API也可以用于对List进行排序
List<Integer> sortedList = list.stream().sorted().collect(Collectors.toList());
Map · 键值对
对 Map 的 Key 进行排序
可以使用 TreeMap 来对 Map 的 Key 排序,TreeMap 是有序的保证,底层采用红黑树实现 可以通过比较器 Comparator 对键进行排序,如果没有指定比较器,则按照键自然顺序排序
Map<String, Integer> map = new HashMap<>();
map.put("c", 3);
map.put("a", 1);
map.put("b", 2);
Map<String, Integer> treeMap = new TreeMap<>(map);
System.out.println(treeMap);
对 Map 的 Value 进行排序
可以先将 Map 转换成 List,再对 List 进行排序即可 在转换过程中,我们可以使用 Map.Entry 类型的元素来表示键值对,然后将该元素添加到 List 中
Map<String, Integer> map = new HashMap<>();
map.put("c", 3);
map.put("a", 1);
map.put("b", 2);
List<Map.Entry<String, Integer>> entryList = new ArrayList<>(map.entrySet());
Collections.sort(entryList, new Comparator<Map.Entry<String, Integer>>() {
@Override
public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {
return Integer.compare(o1.getValue(), o2.getValue());
}
});
// 输出排序后的结果
for (Map.Entry<String, Integer> entry : entryList) {
System.out.println(entry.getKey() + ":" + entry.getValue());
}
Map<String, Person> map2 = new HashMap<>();
map2.put("c", new Person("Tom", 18));
map2.put("a", new Person("Jerry", 20));
map2.put("b", new Person("Alice", 16));
// 将 Map 的 Entry 转换为 List,并按年龄升序排列
List<Map.Entry<String, Person>> list2 = new ArrayList<>(map2.entrySet());
list2.sort((o1, o2) ->
Integer.compare(o1.getValue().getAge(), o2.getValue().getAge()));
// 输出排序后的结果
for (Map.Entry<String, Person> entry : list2) {
System.out.println(entry.getKey() + ":" + entry.getValue());
}
搞定、撒个花( ̄∇ ̄)/🎉~~~~~~~~~~