将原始类型数组的值收集到 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 方法。