實現HashMap的排序

已知一個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;
    }
}


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章