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