已知一个HashMap<Integer, User>集合,User有name和age属性,请写一个方法实现对HashMap的排序功能,该方法接收HashMap<Integer, User>为形参,返回类型为HashMap<Integer, User>,要求对HashMap中的User的age倒序排序,排序时key=value键值对不能拆散。
解析:HashMap本身是不可排序的,但是这道题偏偏让给HashMap排序,那我们就得想在API中有没有这样的Map结构是有序的,LinkedHashMap既是Map结构,也是链表结构,有序的,它也是HashMap的子类,所以我们返回LinkedHashMap<Integer, User>即可,这也符合面向接口(父类编程的思想)。
但凡对集合的操作,我们应保持一个原则就是能用JDK中的API就用JDK中的API,比如排序算法我们不应该去用冒泡或者选择排序,而是首先想到Collections集合工具类。
代码如下:
import java.util.*;
public class Demo{
public static void main(String[] args) {
HashMap<Integer, User> hashMap = new HashMap<>();
hashMap.put(1, new User("张三", 28));
hashMap.put(2, new User("李四", 23));
hashMap.put(3, new User("王五", 26));
hashMap.put(4, new User("赵六", 21));
hashMap.put(5, new User("田七", 29));
hashMap.put(6, new User("钱八", 24));
HashMap<Integer, User> sortMap = sort(hashMap);
System.out.println(sortMap);
}
public static HashMap<Integer, User> sort(HashMap<Integer, User> hashMap) {
//首先获取HashMap的键值对集合
Set<Map.Entry<Integer, User>> entrySet = hashMap.entrySet();
//将set集合转换为list集合,方便使用Collections类排序
List<Map.Entry<Integer, User>> list = new ArrayList<>(entrySet);
//使用Collections集合工具类对list进行排序
Collections.sort(list, (o1, o2) -> {
//按年龄降序
return o2.getValue().getAge() - o1.getValue().getAge();
});
//创建一个有序的HashMap子类集合
LinkedHashMap<Integer, User> linkedHashMap = new LinkedHashMap<>();
//将list集合中的数据存入linkedHashMap
for (Map.Entry<Integer, User> entry : list) {
linkedHashMap.put(entry.getKey(), entry.getValue());
}
return linkedHashMap;
}
}