List和Map的排序

今天说下 List 和 Map 的排序问题，用的比较多，做个记录。

List 的排序

1. 新建 User 类并初始化集合

1.1新建 User 类

@Data
@AllArgsConstructor
class User{
    private String name;
    private int age;
}

1.2 初始化集合

//初始化集合
List<User> list = Lists.newArrayList(
        new User("tom", 12),
        new User("tom", 8),
        new User("sam", 16));

2. 排序

2.1 根据 age 倒序排序

//根据年龄倒序排序
Collections.sort(list, new Comparator<User>(){
    @Override
    public int compare(User o1, User o2) {
        return o2.getAge() - o1.getAge();
    }
});
//打印结果
list.forEach(System.out::println);

查看结果:

User(name=sam, age=16)
User(name=tom, age=12)
User(name=tom, age=8)

结果符合预期，user 的 age 是倒序的

2.2 使用 lambda 表达式的写法会更简洁

//根据年龄倒序排序, 写法 1
//Collections.sort(list, (a,b) -> b.getAge() - a.getAge());
//写法 2
Collections.sort(list, Comparator.comparing(User::getAge, (a,b) -> b - a));
//打印结果
list.forEach(System.out::println);

查看结果：

User(name=tom, age=12)
User(name=tom, age=8)
User(name=sam, age=16)

结果跟上面是一样的。

Map 的排序

1. 按照 key 的排序

1.1 正序排序

//创建 treeMap 对象，默认按照 key 正序排序
Map<String, Integer> map = Maps.newTreeMap();
map.put("b", 2);
map.put("a", 1);
map.put("c", 3);

//打印结果
map.entrySet().forEach(System.out::println);

查看结果：

a=1
b=2
c=3

1.2 倒序排序

//按照 key 倒序排序，自定义 Comparator 对象
Map<String, Integer> map = Maps.newTreeMap(new Comparator<String>() {
    @Override
    public int compare(String o1, String o2) {
        return o2.compareTo(o1);
    }
});
//打印结果
map.entrySet().forEach(System.out::println);

查看结果：

c=3
b=2
a=1

1.3 lambda 表达式的写法

//按照 key 倒序排序，自定义 Comparator 对象
Map<String, Integer> map = new TreeMap<>((a, b) -> b.compareTo(a));

map.put("b", 2);
map.put("a", 1);
map.put("c", 3);

//打印结果
map.entrySet().forEach(System.out::println);

2. 按照 value 排序

2.1 按照 value 倒序排序

//初始化 treeMap 对象
Map<String, Integer> map = new TreeMap<>((a, b) -> b.compareTo(a));
map.put("b", 2);
map.put("a", 1);
map.put("c", 3);
//先将 map 转换为 list，然后调用 sort 方法倒序排序
List<Map.Entry<String, Integer>> list = new ArrayList<>(map.entrySet());
list.sort(list, new Comparator<Map.Entry<String, Integer>>() {
    //倒序排序
    @Override
    public int compare(Map.Entry<String, Integer> o1,
        Map.Entry<String, Integer> o2) {
        return o2.getValue().compareTo(o1.getValue());
    }
});
//打印结果
map.entrySet().forEach(System.out::println);

查看结果：

c=3
b=2
a=1

2.2 lambda 表达式的写法

//初始化 treeMap 对象
Map<String, Integer> map = new TreeMap<>((a, b) -> b.compareTo(a));
map.put("b", 2);
map.put("a", 1);
map.put("c", 3);
//先将 map 转换为 list，然后调用 sort 方法倒序排序
new ArrayList<>(map.entrySet()).sort((a, b) -> b.getValue().compareTo(a.getValue()));
//打印结果
map.entrySet().forEach(System.out::println);

输出结果和上面是一样的。