Java - Sort a TreeMap by Values with Comparator
The collection TreeMap is by default sorted by its keys, but if you need to sort TreeMap by values then Java provides a way to do this using the comparator.
Example:
package codeurjava;
import java.util.*;
class TreeMapSort {
// Sort TreeMap elements by values
public static> Map sortByValues(final Map map) {
ComparatorvalueComparator = new Comparator () {
public int compare(K k1, K k2) {
int compare = map.get(k1).compareTo(map.get(k2));
if (compare == 0)
return 1;
else
return compare;
}
};
MapTreeMap_triee_par_valeur = new TreeMap (valueComparator);
TreeMap_triee_par_valeur.putAll(map);
return TreeMap_triee_par_valeur;
}
public static void main(String args[]) {
TreeMaptreemap = new TreeMap ();
// elements
treemap.put("1", "e4");
treemap.put("2", "e5");
treemap.put("3", "e1");
treemap.put("4", "e3");
treemap.put("5", "e2");
System.out.println("Before sorting:");
/* browse the unsorted TreeMap with Iterator */
Set set = treemap.entrySet();
Iterator i = set.iterator();
// Show elements
while(i.hasNext()) {
Map.Entry entry = (Map.Entry)i.next();
System.out.print(entry.getKey() + " : ");
System.out.println(entry.getValue());
}
System.out.println("After sorting:");
// call the sortByvalues()
Map method TreeMap_triee_par_valeur = sortByValues(treemap);
/* browse the new map sorted with Iterator */
set = TreeMap_triee_par_valeur.entrySet();
i = set.iterator();
// Show elements
while(i.hasNext()) {
Map.Entry entry = (Map.Entry)i.next();
System.out.print(entry.getKey() + " : ");
System.out.println(entry.getValue());
}
}
}
Execution
Before sorting:
1 : e4
2 : e5
3 : e1
4 : e3
5 : e2
After sorting:
3 : e1
5 : e2
4 : e3
1 : e4
2 : e5
References: