今天說下 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 根據(jù) age 倒序排序
//根據(jù)年齡倒序排序
Collections.sort(list, new Comparator<User>(){
@Override
public int compare(User o1, User o2) {
return o2.getAge() - o1.getAge();
}
});
//打印結(jié)果
list.forEach(System.out::println);
查看結(jié)果:
User(name=sam, age=16)
User(name=tom, age=12)
User(name=tom, age=8)
結(jié)果符合預(yù)期,user 的 age 是倒序的
2.2 使用 lambda 表達(dá)式的寫法會更簡潔
//根據(jù)年齡倒序排序, 寫法 1
//Collections.sort(list, (a,b) -> b.getAge() - a.getAge());
//寫法 2
Collections.sort(list, Comparator.comparing(User::getAge, (a,b) -> b - a));
//打印結(jié)果
list.forEach(System.out::println);
查看結(jié)果:
User(name=tom, age=12)
User(name=tom, age=8)
User(name=sam, age=16)
結(jié)果跟上面是一樣的。
Map 的排序
1. 按照 key 的排序
1.1 正序排序
//創(chuàng)建 treeMap 對象,默認(rèn)按照 key 正序排序
Map<String, Integer> map = Maps.newTreeMap();
map.put("b", 2);
map.put("a", 1);
map.put("c", 3);
//打印結(jié)果
map.entrySet().forEach(System.out::println);
查看結(jié)果:
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);
}
});
//打印結(jié)果
map.entrySet().forEach(System.out::println);
查看結(jié)果:
c=3
b=2
a=1
1.3 lambda 表達(dá)式的寫法
//按照 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);
//打印結(jié)果
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 轉(zhuǎn)換為 list,然后調(diào)用 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());
}
});
//打印結(jié)果
map.entrySet().forEach(System.out::println);
查看結(jié)果:
c=3
b=2
a=1
2.2 lambda 表達(dá)式的寫法
//初始化 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 轉(zhuǎn)換為 list,然后調(diào)用 sort 方法倒序排序
new ArrayList<>(map.entrySet()).sort((a, b) -> b.getValue().compareTo(a.getValue()));
//打印結(jié)果
map.entrySet().forEach(System.out::println);
輸出結(jié)果和上面是一樣的。