
引言
在软件开发中,我们经常需要对数据结构进行转换以满足不同的业务需求。一个常见的场景是,我们可能拥有一个表示“项目-用户列表”关系的映射,即一个map
问题描述与目标
假设我们有以下初始数据结构:
Map> map = new HashMap<>(); map.put("projectA", Arrays.asList(new User(1,"Bob"), new User(2,"John"), new User(3,"Mo"))); map.put("projectB", Arrays.asList(new User(2,"John"), new User(3,"Mo"))); map.put("projectC", Arrays.asList(new User(3,"Mo")));
其中,User类定义如下:
public static record User(int id, String name) {}我们的目标是将其转换为以下形式:
Bob = [projectA] John = [projectA, projectB] Mo = [projectA, projectB, projectC]
对应的数据结构为 Map
立即学习“Java免费学习笔记(深入)”;
核心转换思路
实现这种转换的核心思路是遍历原始Map的每一个条目。对于每一个条目,我们获取其键(项目名称)和值(用户列表)。然后,我们遍历这个用户列表中的每一个用户。对于每个用户,我们将其作为新Map的键,并将当前的项目名称添加到该用户对应的项目列表中。
为了确保每个用户在新的Map中都有一个对应的项目列表,我们需要在添加项目名称之前检查该用户是否已存在于新Map中。如果不存在,则需要先为该用户创建一个新的空列表。Java的Map.putIfAbsent()方法非常适合处理这种情况。
代码实现
下面是完整的Java代码示例,展示了如何进行上述转换:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
public class ProjectUserTransformer {
// 定义User记录类,简化equals()和hashCode()的实现
public static record User(int id, String name) {
// record类型会自动生成equals(), hashCode(), toString()方法
}
public static void main(String[] args) {
// 原始数据:项目到用户列表的映射
Map> projectToUsersMap = new HashMap<>();
projectToUsersMap.put("projectA", Arrays.asList(new User(1,"Bob"), new User(2,"John"), new User(3,"Mo")));
projectToUsersMap.put("projectB", Arrays.asList(new User(2,"John"), new User(3,"Mo")));
projectToUsersMap.put("projectC", Arrays.asList(new User(3,"Mo")));
// 目标数据:用户到项目列表的映射
Map> userToProjectsMap = new HashMap<>();
// 遍历原始Map的每一个条目
for (Map.Entry> entry : projectToUsersMap.entrySet()) {
String projectName = entry.getKey(); // 获取当前项目名称
List usersInProject = entry.getValue(); // 获取当前项目中的用户列表
// 遍历当前项目中的每一个用户
for (User user : usersInProject) {
// 如果用户尚未存在于userToProjectsMap中,则为其创建一个新的ArrayList
userToProjectsMap.putIfAbsent(user, new ArrayList<>());
// 将当前项目名称添加到该用户对应的项目列表中
userToProjectsMap.get(user).add(projectName);
}
}
// 打印转换后的结果
System.out.println(userToProjectsMap);
}
} 运行结果示例:
{User[id=1, name=Bob]=[projectA], User[id=2, name=John]=[projectA, projectB], User[id=3, name=Mo]=[projectA, projectB, projectC]}请注意,User[id=2, name=John]对应的项目列表顺序可能因Map的遍历顺序而异,但包含的项目是正确的。
代码解析
-
public static record User(int id, String name) {}:
- Java record类型是一种简洁的类声明方式,特别适合用于不可变的数据载体。
- 它会自动生成构造函数、equals()、hashCode()和toString()方法。这对于将User对象作为Map的键至关重要,因为Map依赖于equals()和hashCode()来正确地存储和检索键。如果User是一个普通类,则需要手动重写这些方法。
-
Map
> projectToUsersMap = new HashMap(); :- 初始化原始的Map,键是String(项目名),值是List
。
- 初始化原始的Map,键是String(项目名),值是List
-
Map
> userToProjectsMap = new HashMap(); :- 初始化用于存储转换结果的新Map,键是User对象,值是List
(项目名列表)。
- 初始化用于存储转换结果的新Map,键是User对象,值是List
-
for (Map.Entry
> entry : projectToUsersMap.entrySet()) :- 这是外层循环,用于遍历projectToUsersMap中的每一个键值对。entry.getKey()将返回项目名称,entry.getValue()将返回该项目下的用户列表。
-
for (User user : usersInProject):
- 这是内层循环,用于遍历当前项目usersInProject中的每一个User对象。
-
userToProjectsMap.putIfAbsent(user, new ArrayList());:
- 这是实现转换的关键一步。putIfAbsent(key, value)方法会检查userToProjectsMap中是否已经存在user作为键。
- 如果user不存在,则将user作为键,并将一个新的空ArrayList
作为值放入userToProjectsMap中。 - 如果user已经存在,则不执行任何操作(即不覆盖已有的值),并返回已存在的值。
- 如果user不存在,则将user作为键,并将一个新的空ArrayList
- 这确保了每个用户在userToProjectsMap中都有一个关联的List
,无论是第一次遇到该用户,还是后续再次遇到。
- 这是实现转换的关键一步。putIfAbsent(key, value)方法会检查userToProjectsMap中是否已经存在user作为键。
-
userToProjectsMap.get(user).add(projectName);:
- 获取user在userToProjectsMap中对应的List
(该列表要么是新创建的,要么是之前已有的)。 - 将当前的项目名称projectName添加到这个列表中。
- 获取user在userToProjectsMap中对应的List
注意事项
- User对象的equals()和hashCode()方法: 正如前文所述,当自定义对象作为Map的键时,正确实现equals()和hashCode()方法至关重要。record类型自动处理了这一点,确保了具有相同id和name的User对象被视为同一个键。如果使用普通类,务必手动重写这两个方法。
-
列表顺序: 转换后的Map中,每个用户对应的项目列表的顺序可能取决于原始Map的遍历顺序以及用户在List
中的顺序。如果需要特定的顺序(例如按字母顺序),则在添加项目名称后或在结果输出前,需要对列表进行排序。 - 性能考量: 对于大规模数据集,这种嵌套循环的方法在大多数情况下是高效且直观的。其时间复杂度大致为O(N*M),其中N是原始Map中的项目数,M是所有项目中的用户总数(或平均每个项目中的用户数)。对于非常庞大的数据集,可以考虑使用并行流(Java Stream API)来进一步优化性能,但这会增加代码的复杂性。
总结
本教程详细介绍了如何将Map










