TreeMap中key和value的排序学习
TreeMap
TreeMap底层是根据红黑树的数据结构构建的,默认是根据key的自然排序来组织(比如integer的大小,String的字典排序),而没有提供根据value进行排序的方法,但又常常遇到需要对value进行排序的情况。
在此,主要目的是记录TreeMap中的key和value的排序实现。
TreeMap底层实现原理
Lambda表达式
在往下进行之前,先了解一个很强大的功能。
自java8发布后,其中一个很重要的新特性 -- Lambda表达式
它允许将行为传入函数中,但在这之前我们想要将行为传入函数时,仅有的选择是匿名内部类(内容相对麻烦些)。
而Lambda表达式的应用,将大量替代匿名内部类的使用,在简化代码的同时,更突出了原来匿名内部类中最重要的那部分包含真正逻辑的代码。
Lambda学习的博客推荐
对Key的排序
由于TreeMap默认就是对Key进行排序的,因此实现起来相对容易,如下
public static void main(String[] args){ //TreeMap<Integer,Integer> map1 = new TreeMap<>(); //默认的TreeMap升序排列 TreeMap<Integer,Integer> map1= new TreeMap<>((Integer a,Integer b) -> b-a); //降序 map1.put(1,2); map1.put(2,4); map1.put(7, 1); map1.put(5,2); System.out.println("map1="+map1); }
对Value的排序
在网上学了一些博客,大致的思路都是先把TreeMap的EntrySet转换为List,然后使用Collections.sort()方法进行排序的。
牛客-名字的漂亮度 :这道题便可应用TreeMap对Value进行排序进行解决。
import java.util.*; public class Main{ public static void main(String[] args){ Scanner input = new Scanner(System.in); while(input.hasNext()){ int n = input.nextInt(); for (int i = 0;i < n;i ++){ TreeMap<Character,Integer> charCount = new TreeMap(); //1. 统计每个字符出现的次数 char[] chars = input.next().toCharArray(); for(char ch : chars){ charCount.put(ch,charCount.getOrDefault(ch,0)+1); } //2. 现将map转换为list,然后通过Lambda表达式进行升序排序 List<Map.Entry<Character, Integer>> list = new ArrayList<>(charCount.entrySet()); Collections.sort(list,(Map.Entry<Character, Integer> o1, Map.Entry<Character, Integer> o2) -> -o1.getValue().compareTo(o2.getValue())); //3. 计算完美值 int maxValue = 26 , sum = 0; for(Map.Entry<Character, Integer> entry : list){ int iter = entry.getValue(); while(iter > 0){ sum += maxValue; iter --; } maxValue --; } System.out.println(sum); } } } }