In Java Language, a TreeMap always stores key-value pairs which are in sorted order on the basis of the key. TreeMap implements the NavigableMap interface and extends AbstractMap class. TreeMap contains unique keys.
Sorting TreeMap by value in Java
- The elements in TreeMap are sorted on the basis of keys.
- So, we need to develop our own logic to sort it on the basis of value. We can do it using comparator class
Example 1:
Java
// Java program to Sort a TreeMap By Value import java.util.*;
class GFG {
public static <K, V extends Comparable<V> > Map<K, V>
valueSort( final Map<K, V> map)
{
// Static Method with return type Map and
// extending comparator class which compares values
// associated with two keys
Comparator<K> valueComparator = new Comparator<K>() {
// return comparison results of values of
// two keys
public int compare(K k1, K k2)
{
int comp = map.get(k1).compareTo(
map.get(k2));
if (comp == 0 )
return 1 ;
else
return comp;
}
};
// SortedMap created using the comparator
Map<K, V> sorted = new TreeMap<K, V>(valueComparator);
sorted.putAll(map);
return sorted;
}
public static void main(String[] args)
{
TreeMap<String, Integer> map = new TreeMap<String, Integer>();
// Put elements to the map
map.put( "Anshu" , 2 );
map.put( "Rajiv" , 4 );
map.put( "Chhotu" , 3 );
map.put( "Golu" , 5 );
map.put( "Sita" , 1 );
// Calling the method valueSort
Map sortedMap = valueSort(map);
// Get a set of the entries on the sorted map
Set set = sortedMap.entrySet();
// Get an iterator
Iterator i = set.iterator();
// Display elements
while (i.hasNext()) {
Map.Entry mp = (Map.Entry)i.next();
System.out.print(mp.getKey() + ": " );
System.out.println(mp.getValue());
}
}
} |
Output
Sita: 1 Anshu: 2 Chhotu: 3 Rajiv: 4 Golu: 5
Example 2:
Java
// Java program to Sort a TreeMap By Value import java.util.*;
class GFG {
// Method for sorting the TreeMap based on values
public static <K, V extends Comparable<V> > Map<K, V>
valueSort( final Map<K, V> map)
{
// Static Method with return type Map and
// extending comparator class which compares values
// associated with two keys
Comparator<K> valueComparator = new Comparator<K>()
{
public int compare(K k1, K k2)
{
int comp = map.get(k1).compareTo(map.get(k2));
if (comp == 0 )
return 1 ;
else
return comp;
}
};
// SortedMap created using the comparator
Map<K, V> sorted = new TreeMap<K, V>(valueComparator);
sorted.putAll(map);
return sorted;
}
public static void main(String[] args)
{
TreeMap<Integer, String> map
= new TreeMap<Integer, String>();
// Put elements to the map
map.put( 1 , "Anshu" );
map.put( 5 , "Rajiv" );
map.put( 3 , "Chhotu" );
map.put( 2 , "Golu" );
map.put( 4 , "Sita" );
// Calling the method valueSort
Map sortedMap = valueSort(map);
// Get a set of the entries on the sorted map
Set set = sortedMap.entrySet();
// Get an iterator
Iterator i = set.iterator();
while (i.hasNext())
{
Map.Entry mp = (Map.Entry)i.next();
System.out.print(mp.getKey() + ": " );
System.out.println(mp.getValue());
}
}
} |
Output
1: Anshu 3: Chhotu 2: Golu 5: Rajiv 4: Sita