17370845950

将原始数组值收集到 Map 集合中

本文介绍了如何将 java 中的原始类型数组(如 `int[]`, `long[]`, `double[]`)转换为 `map` 集合。主要讲解了两种方法:一种是先将原始类型数组“装箱”为对应的包装类型,再使用 `collectors.tomap` 进行收集;另一种是直接利用原始类型流的 `collect` 方法,自定义收集器来实现。通过示例代码,清晰展示了这两种方法的使用和区别,帮助开发者灵活选择适合场景的方案。

在 Java 开发中,我们经常需要将数据从一种形式转换为另一种形式。本文将探讨如何有效地将原始类型数组,例如 int[],long[] 或 double[],转换为 Map 类型的集合。本文将介绍两种主要方法:使用装箱(Boxing)和使用原始类型流(Primitive Streams)。

方法一:使用装箱 (Boxing)

Java 提供了与每种原始类型对应的包装类。例如,int 对应 Integer,double 对应 Double,依此类推。装箱是指将原始类型转换为其对应的包装类。

以下是 Java 中的 8 种原始类型及其对应的包装类:

  • boolean → Boolean
  • byte → Byte
  • char → Character
  • double → Double
  • float → Float
  • int → Integer
  • long → Long
  • short → Short

通过将原始类型数组装箱为包装类型数组,我们可以利用 Stream API 和 Collectors.toMap 方法来创建 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); // {0=zero, 3=three, 5=five, 6=six, 7=seven, 8=eight, 9=nine}
    }
}

代码解释:

  1. Arrays.stream(values): 将 int[] 数组转换为 IntStream。
  2. .boxed(): 将 IntStream 中的每个 int 值装箱为 Integer 对象,生成 Stream
  3. .collect(Collectors.toMap(...)): 使用 Collectors.toMap 收集器将 Stream 转换为 Map
    • Function.identity(): 作为 key 的映射函数,这里直接使用 Integer 本身作为 Key。
    • value -> words[value]: 作为 value 的映射函数,使用数组中的值作为索引,从 words 数组中获取对应的字符串。

方法二:使用原始类型流 (Primitive Streams)

Arrays.stream 方法针对 double[],int[] 和 long[] 进行了重载,分别返回 DoubleStream,IntStream 和 LongStream。 这些原始类型流提供了直接操作原始类型的方法,避免了装箱操作。

示例代码:

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); // {0=zero, 3=three, 5=five, 6=six, 7=seven, 8=eight, 9=nine}
    }
}

代码解释:

  1. Arrays.stream(values): 将 int[] 数组转换为 IntStream。
  2. .collect(supplier, accumulator, combiner): 使用 collect 方法,自定义收集逻辑。
    • HashMap::new: 提供一个 HashMap 作为累加器。
    • (acc, value) -> acc.put(value, words[value]): 累加器函数,将数组中的 value 作为 key,words[value] 作为 value 放入 HashMap 中。
    • HashMap::putAll: 组合器函数,用于合并多个 HashMap。在并行流中,可能存在多个累加器,需要将它们合并成一个最终结果。

总结

本文介绍了两种将原始类型数组转换为 Map 的方法。

  • 装箱 (Boxing): 通过将原始类型装箱为对应的包装类,可以使用 Collectors.toMap 方便地进行收集。 这种方法代码简洁,易于理解。
  • 原始类型流 (Primitive Streams): 直接使用原始类型流的 collect 方法,可以避免装箱操作,潜在地提高性能。 但需要自定义收集逻辑,代码相对复杂。

选择哪种方法取决于具体的使用场景和性能要求。如果对性能要求不高,或者代码可读性更重要,那么装箱方法可能更合适。 如果对性能有较高要求,并且熟悉原始类型流的使用,那么可以考虑使用原始类型流的方法。