在值上排序树形图并将其作为treemap agin返回。

时间:2015-03-02 20:47:59

标签: java treemap key-value-store

我有树形图,我用代码下面的值对它进行排序。如何再次以树形图的形式获得结果?

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;
}

1 个答案:

答案 0 :(得分:2)

您正在创建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));
}