我如何将二维数组收集到映射＆lt; string，set＆lt; gt;＆gt;？

Question

我有两个字符串的数组。

var array = new String[][]{
    {"a", "1"},
    {"a", "2"},
    {"b", "3"},
    ...
};

如何将上述数组收集到map＆lt; string，set＆lt; string＆gt;＆gt;谁的键是每个数组的第一个元素，而值是数组的第二个元素集？

这样我得到了关注地图？

// Map<String, Set<String>>
<"a", ["1, "2"]>,
<"b", ["3"]>,
...

到目前为止，我发现我可以这样对每个数组的第一个元素进行分类。

Arrays.stream(array).collect(Collectors.groupingBy(
        a -> ((String[])a)[0],
        // how can I collect the downstream?
);

Answer 1

您可以使用collector.mapping（）与toset（） downstream：

Collectors.groupingBy(a -> a[0],
    Collectors.mapping(v -> v[1], Collectors.toSet()))

它产生{a = [1，2]，b = [3]} < /代码>。第一个将流元素映射到数组的第二个元素，然后将其作为映射到组密钥的集合收集。

Answer 2

使用collectors.groupingby与collectors.mapping.mapping（）。

public static void main(String[] args) {
    var array = new String[][]{
        {"a", "1"},
        {"a", "2"},
        {"b", "3"}
    };

    Map<String, Set<String>> result =
        Stream.of(array)
            .collect(Collectors.groupingBy(arr -> arr[0],
                Collectors.mapping(arr -> arr[1],
                    Collectors.toSet())));

    System.out.println(result);
}

这样做的另一种方法是使用collector.of（）：

Map<String, Set<String>> result =
    Stream.of(array)
        .collect(Collector.of(
            HashMap::new,
            (Map<String, Set<String>> map, String[] arr) ->
                map.computeIfAbsent(arr[0], k -> new HashSet<>()).add(arr[1]),
            (Map<String, Set<String>> left, Map<String, Set<String>> right) ->
                { left.putAll(right); return left; }));

输出

{a=[1, 2], b=[3]}

Answer 3

您需要一个collector.mapping（也无需指定string []内部）

var array = new String[][]{{"a", "1"}, {"a", "2"}, {"b", "3"},};

Map<String, Set<String>> res = Arrays.stream(array).collect(Collectors.groupingBy(
        a -> a[0],
        Collectors.mapping(a -> a[1], Collectors.toSet())
));

System.out.println(res); // {a=[1, 2], b=[3]}

Answer 4

您可以简单地使用for-east循环

   Map<String, Set<String>> map=new HashMap<>();
   for(String []arr : array){
     Set<String> set = map.getOrDefault(arr[0], new HashSet<>());
     set.add(arr[1]);
     map.put(arr[0],set);
   }
   System.out.println(map);
 

输出：

{a = [1，2]，b = [3]}

Answer 5

您可以使用collector.tomap而不是collector.groupingby。

public class Main {

  public static void main(String[] args) throws Exception {
    String[][] array = new String[][]{{"a", "1"}, {"a", "2"}, {"b", "3"}};
    Map<String, Set<String>> map = Arrays.stream(array).collect(Collectors.toMap(
            arr -> arr[0],
            arr -> new HashSet<>(Collections.singleton(arr[1])),
            (l, r) -> {
              l.addAll(r);
              return l;
            }));
    System.out.println(map);
  }
}

第一个参数为keymapper，第二个是valuemapper，第三 - MergeFunction。

Answer 6

您可以循环遍历数组，对于每个字符，您可以检查此键是否保存在地图之前，如果是的，则只需将相应的字符推到该重复键的列表中