17370845950

Java中如何用flatMap扁平化嵌套集合
答案:flatMap用于将嵌套集合展开为单层集合。通过将每个元素转换为流并合并,实现多层结构的扁平化,如List转为List,每调用一次flatMap降一级嵌套。

在Java中,当你处理嵌套集合(比如 List>)时,常常需要把多层结构“拍平”成单层集合。这时就可以使用 flatMap 方法来实现扁平化操作。

什么是 flatMap?

flatMap 是 Java 8 引入的 Stream API 中的一个方法,它能将每个元素映射为一个流,然后把这些流合并成一个单独的流。与 map 不同的是,map 只做转换,而 flatMap 会“展平”结果。

基本用法:扁平化 List>

假设你有一个包含多个列表的列表,你想把它变成一个单一的列表:

List> nestedList = Arrays.asList(
    Arrays.asList("a", "b"),
    Arrays.asList("c", "d"),
    Arrays.asList("e", "f")
);

List flatList = nestedList.stream()
    .flatMap(List::stream)
    .collect(Collectors.toList());

System.out.println(flatList); // 输出: [a, b, c, d, e, f]

这里的关键是 .flatMap(List::stream) —— 它把每一个内层 List 转换成一个 Stream,然后把这些 Stream 合并成一个整体流。

更复杂的嵌套结构(如对象中的集合)

常见场景是有一个对象列表,每个对象内部又包含一个集合。例如:

class Department {
    private List employees;

    public Department(List employees) {
        this.employees = employees;
    }

    public List getEmployees() {
        return employees;
    }
}

List departments = Arrays.asList(
    new Department(Arrays.asList("Alice", "Bob")),
    new Department(Arrays.asList("Charlie", "David"))
);

List allEmployees = departments.stream()
    .flatMap(dept -> dept.getEmployees().stream())
    .collect(Collectors.toList());

System.out.println(allEmployees); // [Alice, Bob, Charlie, David]

通过 flatMap(dept -> dept.getEmployees().stream()),我们将每个部门的员工列表展开并合并。

多层嵌套怎么办?

如果嵌套超过两层,比如 List>>,可以连续使用 flatMap:

List>> veryNested = ...;

List result = veryNested.stream()
    .flatMap(List::stream)        // 第一层扁平化
    .flatMap(List::stream)        // 第二层扁平化
    .collect(Collectors.toList());

基本上就这些。flatMap 的核心作用就是“映射 + 扁平化”,特别适合处理层级结构的数据。只要记住:每次 flatMap 处理一层嵌套,调用一次就降一级。