java-8 - 翻转键，使用Java8 Stream在HashMap中的值

我在以下结构中有一个Map，我想翻转键和值。

Map<String, List<String>> dataMap

样本数据 :

acct01: [aa, ab, ad],
acct02: [ac, ad]
acct03: [ax, ab]

希望将此数据转换为

aa: [acct01],
ab: [acct01, acct03],
ac: [acct02],
ad: [acct01, acct02],
ax: [acct03]

想知道是否存在Java 8流方法来转换Map。

我当前的实现(没有Stream)

Map<String, List<String>> originalData = new HashMap<String, List<String>>();
        originalData.put("Acct01", Arrays.asList("aa", "ab", "ad"));
        originalData.put("Acct02", Arrays.asList("ac", "ad"));
        originalData.put("Acct03", Arrays.asList("ax", "ab"));

        System.out.println(originalData);
        Map<String, List<String>> newData = new HashMap<String, List<String>>();
        originalData.entrySet().forEach(entry -> {
            entry.getValue().forEach(v -> {
                if(newData.get(v) == null) {
                    List<String> t = new ArrayList<String>();
                    t.add(entry.getKey());
                    newData.put(v, t);
                } else {
                    newData.get(v).add(entry.getKey());
                }
            });
        });
        System.out.println(newData);

输入和输出{Acct01=[aa, ab, ad], Acct02=[ac, ad], Acct03=[ax, ab]}{aa=[Acct01], ab=[Acct01, Acct03], ac=[Acct02], ad=[Acct01, Acct02], ax=[Acct03]}
寻找使用Stream实现的方法。

最佳答案

获取条目集的流，按键值对将其展平为一个条目(按值分组)，并将关联的键收集到一个列表中。

import static java.util.Arrays.asList;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.mapping;
import static java.util.stream.Collectors.toList;

import java.util.AbstractMap.SimpleImmutableEntry;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

<K, V> Map<V, List<K>> invert(Map<K, List<V>> map) {
    return map.entrySet()
              .stream()
              .flatMap(entry ->
                  entry.getValue()
                       .stream()
                       .map(value ->
                           new SimpleImmutableEntry<>(
                               entry.getKey(),
                               value
                           )
                       )
              )
              .collect(
                  groupingBy(
                      Entry::getValue,
                      mapping(
                          Entry::getKey,
                          toList()
                      )
                  )
              );
}