java - 根据值对树形图进行排序，然后将其作为树形图返回。

Question

我有树形图，我使用下面的代码对其值进行排序。我怎样才能再次获得树形图的结果？

static <K, V extends Comparable<? super V>> SortedSet<Map.Entry<K, V>> entriesSortedByValues(
        Map<K, V> map) {
    SortedSet<Map.Entry<K, V>> sortedEntries = new TreeSet<Map.Entry<K, V>>(
            new Comparator<Map.Entry<K, V>>() {
                @Override
                public int compare(Map.Entry<K, V> e1, Map.Entry<K, V> e2) {
                    int res = e1.getValue().compareTo(e2.getValue());
                    return res != 0 ? res : 1;
                }
            });
    sortedEntries.addAll(map.entrySet());
    return sortedEntries;
}

Answer 1

您正在创建一个TreeSet，而您需要创建一个TreeMap。您传递给 TreeMap 的比较器将使用作为参数传递的 map 来获取值并进行比较。

将方法更改为:

static <K, V extends Comparable<? super V>> TreeMap<K, V> entriesSortedByValues(final Map<K, V> map) {
    TreeMap<K, V> sortedEntries = new TreeMap<K, V>(new Comparator<K>() {
      @Override
      public int compare(K o1, K o2) {
        return map.get(o1).compareTo(map.get(o2));
      }
    });
    sortedEntries.putAll(map);
    return sortedEntries;
}

测试使用:

public static void main(String args[]) {
  Map<Integer, Integer> map = new HashMap<Integer, Integer>();
  map.put(1, 3);
  map.put(3, 1);
  map.put(5, 6);
  map.put(2, 10);
  // Prints: {3=1, 1=3, 5=6, 2=10}
  System.out.println(entriesSortedByValues(map));
}