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


京公网安备 11010502036488号