从流中收集到多图

Question

我有一个短语，我在空白处分割并创建一个地图来保存单词和单词的索引位置。

我工作得很好。但问题是当短语包含重复的单词时。所以我想用com.google.common.collect.Multimap。

有没有办法使用以下流的收集器生成 Multimap？

List<String> words = new ArrayList<>(Arrays.asList(phrase.split("\\s+")));
Map<String, Integer> tokenMap = IntStream.range(0, words.size())
            .boxed()
            .collect(Collectors.toMap(words::get, i -> i));

Answer 1

你可以用这个：

List<String> words = Arrays.asList("a b c d e a b c".split("\\s+"));
Multimap<String, Integer> tokenMap = IntStream.range(0, words.size()).boxed()
        .collect(ArrayListMultimap::create, (m, i) -> m.put(words.get(i), i), Multimap::putAll);

结果将是：

{a=[0, 5], b=[1, 6], c=[2, 7], d=[3], e=[4]}

Answer 2

Stream.collect() 是一个非常通用的 API，所以没有什么特别的原因不能收集到 Multimap。但是Collectors.toMap() 的任何变体都不能支持这一点，因为com.google.common.collect.Multimap 不是java.util.Map。

我不知道有一个现有的Collector 实现可以完成这项工作，但肯定可以通过直接实现Collector 来编写一个。