已知一個HashMap<Integer, User>集合, User有name (String) 和 age (int) 屬性,請寫一個方法實現對HashMap的排序功能,該方法接受HashMap<Integer, User>爲形參,返回類型爲HashMap<Integer, User>,要求對HashMap中的User的age倒序進行排序,排序時key==value鍵值對不能拆散。
注意:
HashMap本身是不支持排序的,但是該道題偏偏需要排序,那我們想API中是否有Map結構而且是有序的,LinkedHashMap, 其是Map也是基於鏈表結構,有序的,並且還是HashMap的子類,我們返回LinkedHashMap<Integer, User>即可。
但凡是對集合的操作,我們應該保持一個原則就是能用JDK中的API就用,比如排序算法不選用冒泡或者選擇,而是首先想到用Collections集合的工具類。
public class SortTest {
public static void main(String[] args) {
HashMap<Integer, User> map = new HashMap<Integer, User>();
map.put(1, new User("狗娃", 23));
map.put(2, new User("鐵蛋", 18));
map.put(3, new User("狗蛋", 26));
System.out.println(map);
HashMap<Integer, User> sortHashMap = sortHashMap(map);
System.out.println(sortHashMap);
}
public static HashMap<Integer, User> sortHashMap(HashMap<Integer,User> map){
//先拿到map的鍵值集合
Set<Map.Entry<Integer, User>> entrySet = map.entrySet();
//利用一個set集合構造一個 list集合 主要是爲了使用Collections的排序方法
List<Map.Entry<Integer, User>> list = new ArrayList<Map.Entry<Integer, User>>(entrySet);
//使用Collections集合工具類對List進行排序
Collections.sort(list, new Comparator<Map.Entry<Integer, User>>() {
@Override
//實現倒序排序
public int compare(Map.Entry<Integer, User> o1, Map.Entry<Integer, User> o2) {
return o2.getValue().getAge()-o1.getValue().getAge();
}
});
//創建一個新的有序的HashMap子類集合
LinkedHashMap<Integer, User> linkedHashMap = new LinkedHashMap<>();
for (Map.Entry<Integer,User> entry : list){
//將list中的數據存儲到新集合中
linkedHashMap.put(entry.getKey(),entry.getValue());
}
//返回集合
return linkedHashMap;
}
}