TreeMap的排序

概述

Map是键值对的集合接口，它的实现类主要包括：HashMap、Hashtable以及LinkedHashMap等。

1. TreeMap：基于红黑树（Red-Black tree）的 NavigableMap 实现，该映射根据其键的自然顺序进行排序，或者根据创建映射时提供的 Comparator 进行排序，具体取决于使用的构造方法。
2. HashMap的值是没有顺序的，它是按照key的HashCode来实现的，对于这个无序的HashMap我们要怎么来实现排序呢？参照TreeMap的value排序。

Map.Entry返回Collections视图。

按key排序

TreeMap默认是升序的，如果我们需要改变排序方式，则需要使用比较器：Comparator。Comparator可以对集合对象或者数组进行排序的比较器接口，实现该接口的public compare(T o1,To2)方法即可实现排序。

TreeMap默认按key升序排序

public class T0 {
    public static void main(String[] args) {
        TreeMap<Integer, String> treeMap = new TreeMap<>();
        treeMap.put(1, "a");
        treeMap.put(3, "b");
        treeMap.put(2, "c");

        for (Map.Entry<Integer, String> item :
                treeMap.entrySet()) {
            System.out.println(item);
        }
        Map.Entry<Integer, String> entry = treeMap.firstEntry();
        System.out.println("entry = " + entry);
    }
}

修改TreeMap的排序方式，按key降序排序

实现方式是自定义Comparator,可以直接使用Comparator.reverseOrder()或者手动实现Comparator.

new TreeMap<>(Comparator.reverseOrder())

public class T1 {
    public static void main(String[] args) {
        Comparator<Integer> keyComparator = new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2.compareTo(o1);
            }
        };

        TreeMap<Integer, String> treeMap = new TreeMap(keyComparator);
        treeMap.put(1, "a");
        treeMap.put(3, "b");
        treeMap.put(2, "c");

        for (Map.Entry<Integer, String> item :
                treeMap.entrySet()) {
            System.out.println(item);
        }
        Map.Entry<Integer, String> entry = treeMap.firstEntry();
        System.out.println("entry = " + entry);
    }
}

按Value排序

以下只演示按TreeMap按Value升序排序，这同样适用于HashMap。

修改TreeMap的排序方式，按Value升序排序

注意：正常情况下Map是不可以使用Collections.sort()方法进行排序的，不过可以将Map转换成list之后再进行排序。

public class T2 {
    public static void main(String[] args) {

        Comparator<Map.Entry<Integer, String>> comparator = new Comparator<Map.Entry<Integer, String>>() {
            @Override
            public int compare(Map.Entry<Integer, String> o1, Map.Entry<Integer, String> o2) {
                return o1.getValue().compareTo(o2.getValue());
            }
        };

        TreeMap<Integer, String> treeMap = new TreeMap<>();
        treeMap.put(1, "a");
        treeMap.put(9, "e");
        treeMap.put(3, "b");
        treeMap.put(2, "c");
        ArrayList<Map.Entry<Integer, String>> list = new ArrayList<Map.Entry<Integer, String>>(treeMap.entrySet());
        Collections.sort(list, comparator);

        for (Map.Entry<Integer, String> entry : list) {
            System.out.println(entry);
        }
    }
}