如何使用Java流将两个数组合并到一个映射中？

Question

让我们假设我们得到了以下两个数组

String[] keys   = new String[] {"a", "b", "c", "aa", "d", "b"}
int[]    values = new int[]    { 1 ,  2 ,  3 ,  4  ,  5 ,  6 }

通过将这两个数组合并到HashTable中，我们得到以下结果

// pseudo-code
Map<String, Integer> dictionary = new HashTable<>(
  ("a"  => 1)
  ("b"  => 8) // because "b" appeared in index 1 and 5
  ("c"  => 3)
  ("aa" => 4)
  ("d"  => 5)
);

我们怎么能用java Lambda样式做到这一点？

到目前为止，我有以下内容：

// this loops through the range given (used for index start and end)
// and sums the values of duplicated keys
tree.listMap = IntStream.range(range[0], range[1]).boxed().collect(
  Collectors.toMap(i - > dictionary[i], i - > i + 1, Integer::sum, TreeMap::new)
);

但是，我想采用2个数组，按键和值合并它们，其中value是重复键的所有值的总和。 我们应该怎么做？

Answer 1

你去：

Map<String,Integer> themap = 
       IntStream.range (0, keys.length).boxed()
                .collect (Collectors.toMap(i->keys[i],
                                           i->values[i],
                                           Integer::sum,
                                           TreeMap::new));

输出：

{a=1, aa=4, b=8, c=3, d=5}

这与您发布的代码段非常相似，但由于某些原因，您发布的代码段不包含对keys和values数组的引用。

Answer 2

我不喜欢在引用索引时使用流，但是你可以使用groupingBy和summingInt来完成这个：

Map<String, Integer> result = IntStream.range(0, keys.length)
   .boxed()
   .collect(
       Collectors.groupingBy(
           i -> keys[i],
           Collectors.summingInt(i -> values[i])
       )
   );

请注意，这可以假设键和值都具有相同的长度，因此您可能需要进行一些额外的验证。