如何获取TreeMap<String, ArrayList<CustomModel>>中存储的List对象？

一、如何从TreeMap<String, ArrayList<CustomModel>>中获取List对象？ #

其实和普通Map的获取逻辑一致，TreeMap本身实现了Map接口，常用的获取方式都适用：

1. 通过指定key直接获取 #

如果你知道目标类别的key，直接调用get()方法就能拿到对应的ArrayList：

// 假设你的TreeMap实例名为sortedCategoryMap
String targetCategory = "Books";
ArrayList<CustomModel> modelList = sortedCategoryMap.get(targetCategory);
// 注意：若key不存在，get()会返回null，建议先做非空判断
if (modelList != null) {
    // 处理列表逻辑
}

2. 遍历所有类别对应的List #

如果需要处理所有类别下的List，用entrySet遍历效率更高（避免重复调用get()）：

for (Map.Entry<String, ArrayList<CustomModel>> entry : sortedCategoryMap.entrySet()) {
    String category = entry.getKey();
    ArrayList<CustomModel> modelList = entry.getValue();
    // 处理当前类别的列表
    System.out.printf("类别：%s，包含%d个对象%n", category, modelList.size());
}

3. 合并所有List为一个大列表 #

如果需要把所有类别的对象合并成一个列表，可以用Stream简化操作：

List<CustomModel> allModels = sortedCategoryMap.values().stream()
    .flatMap(List::stream)
    .collect(Collectors.toList());

二、更优的实现方案：一步完成分组+排序 #

你当前先转HashMap再转TreeMap的方式是可行的，但其实可以用Java Stream的Collectors.groupingBy()直接指定TreeMap作为容器，一步完成按类别分组+类别自然字母排序，避免中间HashMap的额外开销，代码更简洁：

示例代码 #

// 假设原始列表为originalList
TreeMap<String, ArrayList<CustomModel>> sortedCategoryMap = originalList.stream()
    .collect(Collectors.groupingBy(
        CustomModel::getCategory, // 按CustomModel的category字段分组
        TreeMap::new, // 指定用TreeMap存储，自动按key自然排序
        Collectors.toCollection(ArrayList::new) // 指定分组内元素存在ArrayList中
    ));

额外优化：分组内元素也按字母排序 #

如果不仅类别要排序，每个类别下的CustomModel对象也需要按某个字段（比如名称）排序，可以在收集时直接处理：

TreeMap<String, List<CustomModel>> sortedCategoryMap = originalList.stream()
    .sorted(Comparator.comparing(CustomModel::getName)) // 先按对象名称排序
    .collect(Collectors.groupingBy(
        CustomModel::getCategory,
        TreeMap::new,
        Collectors.toList()
    ));

或者分组后单独排序：

sortedCategoryMap.forEach((category, list) -> {
    list.sort(Comparator.comparing(CustomModel::getName));
});

为什么这个方案更优？ #

• 减少中间集合创建，节省内存和GC开销
• 代码逻辑连贯，可读性更强，符合现代Java写法
• 直接利用Stream API的特性，避免手动转换集合的繁琐步骤

内容的提问来源于stack exchange，提问作者portfoliobuilder