以数组列表为值的Java 8 Collectors.toMap

Question

我想使用java 8将给定的二维数组转换为map。输入 - { { 1,0}，{2,0}，{3,1}，{3,2}}输出应为Map形式

以下是我的解决方案

for (int[] prereq : prerequisites) {
            map.computeIfAbsent(prereq[0], k -> new ArrayList<>()).add(prereq[1]);
        }

有没有更好的方法，如果for循环可以用streams代替。

Answer 1

由于数组行始终由一对数字组成，并且您将第一个元素视为键，将第二个元素视为值，因此乍一看，如果将它们映射到 SimpleEntry，则很明显：

int[][] prerequisites = { { 1, 0 }, { 2, 0 }, { 3, 1 }, { 3, 2 } };
    Map<Integer,List<Integer>> map = 
            Arrays.stream(prerequisites)
                  .map(arr -> new AbstractMap.SimpleEntry<>(arr[0], arr[1]))
                  .collect(Collectors.groupingBy(
                            Map.Entry::getKey,
                            Collectors.mapping(Map.Entry::getValue, Collectors.toList())));

Answer 2

要通过键收集到包含多个值的映射中，请使用< code > collectors . grouping by 。

int[][] prerequisites = {{1, 0}, {2, 0}, {3, 1}, {3, 2}};
Map<Integer, List<Integer>> res = Arrays.stream(prerequisites).collect(
        Collectors.groupingBy(x -> x[0], Collectors.mapping(x -> x[1], Collectors.toList())));

输出：

{1=[0], 2=[0], 3=[1, 2]}