TreeMap自定义排序

我们都知道TreeMap可以根据key按字典升序排序。但在某些场景下,我们需要自定义排序规则,为了代码优雅一些,我们也希望在stream中groupingBy时自定义排序规则,就可以参考本文的实现。

1. 使用TreeMap默认的排序规则(按字典升序排序)

先来看一下默认的排序规则。

java 复制代码
/**
 * 使用TreeMap默认的排序规则(按字典升序排序)
 */
private static void sortByDefault() {
    TreeMap<String, Integer> treeMap = new TreeMap<>();
    for (int index = 0; index < 5; index++) {
        treeMap.put("str_" + index, index);
    }
    System.out.println("*****TreeMap默认排序规则*****");
    treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
    System.out.println();
}

2. 使用自定义排序规则

使用自定义排序规则对TreeMap进行实例化。

java 复制代码
/**
 * 使用自定义排序规则
 */
private static void sortByCustom() {
    Comparator<String> comparator = (String s1, String s2) -> SortEnum.getSortWeightByValue(s1) - SortEnum.getSortWeightByValue(s2);
    TreeMap<String, Integer> treeMap = new TreeMap<>(comparator);
    for (int index = 0; index < 5; index++) {
        treeMap.put("str_" + index, index);
    }
    System.out.println("*****TreeMap自定义排序规则*****");
    treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
    System.out.println();
}

3. GroupingBy时使用自定义排序规则

一开始直接用TreeMap::new作为groupingBy方法的第二个参数,但是这种办法只能引用TreeMap的无参构造方法。查看groupingBy的入参可以发现,第二个参数是Supplier类型,于是手动实现Supplier,在方法体中传入自定义比较器,然后返回TreeMap对象。

java 复制代码
/**
 * GroupingBy时使用自定义排序规则
 */
private static void sortWhenGroupingBy() {
    List<String> list = new ArrayList<>();
    for (int index = 0; index < 5; index++) {
        list.add("str_" + index);
    }
    Comparator<String> comparator = (s1, s2) -> SortEnum.getSortWeightByValue(s1) - SortEnum.getSortWeightByValue(s2);
    // 如果现在外面定义好Supplier,需要声明TreeMap的key和value的类型,否则可能出现类型不兼容的异常
    // Supplier<TreeMap<String, List<String>>> supplier = () -> new TreeMap<>(comparator);
    // TreeMap<String, List<String>> treeMap = list.stream().collect(Collectors.groupingBy(Function.identity(), supplier, Collectors.toList()));
    TreeMap<String, List<String>> treeMap = list.stream().collect(Collectors.groupingBy(Function.identity(), () -> new TreeMap<>(comparator), Collectors.toList()));
    System.out.println("*****GroupingBy时自定义排序规则*****");
    treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
    System.out.println();
}

完整代码

自定义排序权重类

java 复制代码
public enum SortEnum {
    ZERO(0, "str_1"),
    FIRST(1, "str_0"),
    SECOND(2, "str_3"),
    THREE(3, "str_2"),
    ;
    private int sortWeight;
    private String value;

    SortEnum(int sortWeight, String value) {
        this.sortWeight = sortWeight;
        this.value = value;
    }

    public static int getSortWeightByValue(String value) {
        for (SortEnum sortEnum : SortEnum.values()) {
            if (sortEnum.value.equals(value)) {
                return sortEnum.sortWeight;
            }
        }
        return Integer.MAX_VALUE;
    }
}

测试代码

java 复制代码
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.TreeMap;
import java.util.function.Function;
import java.util.stream.Collectors;

public class TestTreeMap {
    public static void main(String[] args) {
        sortByDefault();
        sortByCustom();
        sortWhenGroupingBy();
    }

    /**
     * 使用TreeMap默认的排序规则(按字典升序排序)
     */
    private static void sortByDefault() {
        TreeMap<String, Integer> treeMap = new TreeMap<>();
        for (int index = 0; index < 5; index++) {
            treeMap.put("str_" + index, index);
        }
        System.out.println("*****TreeMap默认排序规则*****");
        treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
        System.out.println();
    }

    /**
     * 使用自定义排序规则
     */
    private static void sortByCustom() {
        Comparator<String> comparator = (String s1, String s2) -> SortEnum.getSortWeightByValue(s1) - SortEnum.getSortWeightByValue(s2);
        TreeMap<String, Integer> treeMap = new TreeMap<>(comparator);
        for (int index = 0; index < 5; index++) {
            treeMap.put("str_" + index, index);
        }
        System.out.println("*****TreeMap自定义排序规则*****");
        treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
        System.out.println();
    }

    /**
     * GroupingBy时使用自定义排序规则
     */
    private static void sortWhenGroupingBy() {
        List<String> list = new ArrayList<>();
        for (int index = 0; index < 5; index++) {
            list.add("str_" + index);
        }
        Comparator<String> comparator = (s1, s2) -> SortEnum.getSortWeightByValue(s1) - SortEnum.getSortWeightByValue(s2);
        // 如果现在外面定义好Supplier,需要声明TreeMap的key和value的类型,否则可能出现类型不兼容的异常
        // Supplier<TreeMap<String, List<String>>> supplier = () -> new TreeMap<>(comparator);
        // TreeMap<String, List<String>> treeMap = list.stream().collect(Collectors.groupingBy(Function.identity(), supplier, Collectors.toList()));
        TreeMap<String, List<String>> treeMap = list.stream().collect(Collectors.groupingBy(Function.identity(), () -> new TreeMap<>(comparator), Collectors.toList()));
        System.out.println("*****GroupingBy时自定义排序规则*****");
        treeMap.forEach((k, v) -> System.out.println(k + " : " + v));
        System.out.println();
    }
}

运行结果:

复制代码
*****TreeMap默认排序规则*****
str_0 : 0
str_1 : 1
str_2 : 2
str_3 : 3
str_4 : 4

*****TreeMap自定义排序规则*****
str_1 : 1
str_0 : 0
str_3 : 3
str_2 : 2
str_4 : 4

*****GroupingBy时自定义排序规则*****
str_1 : [str_1]
str_0 : [str_0]
str_3 : [str_3]
str_2 : [str_2]
str_4 : [str_4]
相关推荐
HanhahnaH4 分钟前
Spring集合注入Bean
java·spring
未定义.22110 分钟前
电子削铅笔刀顺序图详解:从UML设计到PlantUML实现
java·软件工程·uml
雾月5527 分钟前
LeetCode 1292 元素和小于等于阈值的正方形的最大边长
java·数据结构·算法·leetcode·职场和发展
24k小善1 小时前
Flink TaskManager详解
java·大数据·flink·云计算
想不明白的过度思考者2 小时前
Java从入门到“放弃”(精通)之旅——JavaSE终篇(异常)
java·开发语言
.生产的驴2 小时前
SpringBoot 封装统一API返回格式对象 标准化开发 请求封装 统一格式处理
java·数据库·spring boot·后端·spring·eclipse·maven
猿周LV2 小时前
JMeter 安装及使用 [软件测试工具]
java·测试工具·jmeter·单元测试·压力测试
晨集2 小时前
Uni-App 多端电子合同开源项目介绍
java·spring boot·uni-app·电子合同
时间之城2 小时前
笔记:记一次使用EasyExcel重写convertToExcelData方法无法读取@ExcelDictFormat注解的问题(已解决)
java·spring boot·笔记·spring·excel
椰羊~王小美2 小时前
LeetCode -- Flora -- edit 2025-04-25
java·开发语言