我有树形图,我用代码下面的值对它进行排序。如何再次以树形图的形式获得结果?
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;
}
答案 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));
}