数组值收集到 Map 的方法

本文介绍了如何将 Java 中的原始类型数组（如 `int[]`, `long[]`, `double[]`）转换为 `Map` 集合。主要讲解了两种方法：一种是通过将原始类型“装箱”为对应的包装类，然后使用 `Collectors.toMap` 进行收集；另一种是直接使用原始类型流的 `collect` 方法，自定义收集器的行为。通过示例代码，详细展示了两种方法的实现方式及各自的特点。

在 Java 中，将原始类型数组转换为 Map 集合是一个常见的需求。由于原始类型（如 int, long, double）与对象类型（如 Integer, Long, Double）在 Stream 处理上存在差异，因此需要采用不同的方法来实现。本文将详细介绍两种常用的方法：使用装箱（Boxing）和使用原始类型流的 collect 方法。

方法一：使用装箱（Boxing）

装箱是指将原始类型转换为对应的包装类。Java 提供了自动装箱和拆箱的机制，方便原始类型和包装类之间的转换。

1. 创建原始类型数组的 Stream： 使用 Arrays.stream() 方法将原始类型数组转换为对应的 Stream。例如，int[] 转换为 IntStream。
2. 装箱： 调用 boxed() 方法将原始类型 Stream 转换为对象类型的 Stream。例如，IntStream 转换为 Stream。
3. 使用 Collectors.toMap 收集： 使用 Collectors.toMap() 方法将 Stream 收集到 Map 中。Collectors.toMap() 接收两个 Function 参数，分别用于指定 Key 和 Value 的生成方式。

以下是一个示例代码，演示如何将 int[] 转换为 Map：

import java.util.*;
import java.util.function.*;
import java.util.stream.*;

public class PrimitiveStreamCollection {
    private static final String[] words = {
        "zero", "one", "two", "three", "four",
        "five", "six", "seven", "eight", "nine"
    };

    private static final int[] numbers = { 8, 6, 7, 5, 3, 0, 9 };

    public static Map collectBoxed(int[] values) {
        return Arrays.stream(values)
            .boxed()
            .collect(
                Collectors.toMap(
                    Function.identity(),
                    value -> words[value]));
    }

    public static void main(String[] args) {
        Map boxedMap = collectBoxed(numbers);
        System.out.println(boxedMap);
    }
}

代码解释：

• Arrays.stream(values): 将 int[] 数组 values 转换为 IntStream。
• .boxed(): 将 IntStream 转换为 Stream，即进行装箱操作。
• Collectors.toMap(Function.identity(), value -> words[value]): 使用 Collectors.toMap 收集流。
  • Function.identity(): 使用 Integer 本身作为 Key。
  • value -> words[value]: 使用数组 words 中对应索引的字符串作为 Value。

方法二：使用原始类型流的 collect 方法

IntStream, LongStream, DoubleStream 等原始类型流提供了 collect 方法，该方法允许自定义收集器的行为，无需进行装箱操作，从而提高性能。

collect 方法接收三个参数：

1. Supplier： 用于创建结果容器的 Supplier。例如，HashMap::new 创建一个新的 HashMap 实例。
2. Accumulator： 用于将元素添加到结果容器的 BiConsumer。例如，(acc, value) -> acc.put(value, words[value]) 将 value 作为 Key，words[value] 作为 Value 添加到 acc (HashMap) 中。
3. Combiner： 用于合并多个结果容器的 BinaryOperator。例如，HashMap::putAll 将一个 HashMap 的内容添加到另一个 HashMap 中（在并行流中会用到）。

以下是一个示例代码，演示如何使用原始类型流的 collect 方法将 int[] 转换为 Map：

import java.util.*;
import java.util.function.*;
import java.util.stream.*;

public class PrimitiveStreamCollection {
    private static final String[] words = {
        "zero", "one", "two", "three", "four",
        "five", "six", "seven", "eight", "nine"
    };

    private static final int[] numbers = { 8, 6, 7, 5, 3, 0, 9 };

    public static Map collectUnboxed(int[] values) {
        return Arrays.stream(values)
            .collect(
                HashMap::new,
                (acc, value) -> acc.put(value, words[value]),
                HashMap::putAll);
    }

    public static void main(String[] args) {
        Map unboxedMap = collectUnboxed(numbers);
        System.out.println(unboxedMap);
    }
}

代码解释：

• Arrays.stream(values): 将 int[] 数组 values 转换为 IntStream。
• collect(HashMap::new, (acc, value) -> acc.put(value, words[value]), HashMap::putAll): 使用 collect 方法进行收集。
  • HashMap::new: 创建一个新的 HashMap 作为结果容器。
  • (acc, value) -> acc.put(value, words[value]): 将 value 作为 Key，words[value] 作为 Value 添加到 HashMap 中。
  • HashMap::putAll: 如果使用并行流，则用于合并多个 HashMap。

总结

本文介绍了两种将原始类型数组转换为 Map 集合的方法：使用装箱和使用原始类型流的 collect 方法。

• 装箱： 代码简洁易懂，适用于简单的转换场景。但由于涉及到装箱和拆箱操作，性能相对较低。
• 原始类型流的 collect 方法： 性能更高，适用于需要处理大量数据的场景。但代码相对复杂，需要理解 Supplier、Accumulator 和 Combiner 的作用。

在实际应用中，应根据具体的需求和性能要求选择合适的方法。如果对性能要求不高，可以使用装箱方法；如果需要处理大量数据，建议使用原始类型流的 collect 方法。