Question

我有以下两个String列表：

{APPLE, ORANGE, BANANA} //call it keyList
{APPLE123, ORANGEXXX, 1APPLE, APPLEEEE} //call it valueList

所需的输出是这样的HashMap<String, List<String>>：

<APPLE, {APPLE123, 1APPLE, APPLEEEE}>
<ORANGE, {ORANGEXXX}>
<BANANA, {}> //also <key, null> is accepted

我已经实现了此解决方案（有效）

HashMap<String, List<String>> myMap = new HashMap<>();
keyList.forEach(key -> {
    List<String> values = valueList.stream()
            .filter(value -> value.contains(key))
            .collect(Collectors.toList());
    myMap.put(key, values);
});

假设值仅与一个键相关（这是我的域的约束），那么就性能和/或代码清除而言，这是java8中的最佳解决方案吗？可以以某种方式对其进行调整吗？

Answer 1

如果可以安全地假设每个值都与一个键相关联，并且只有一个键，则可以进入以下方向：

Pattern p = Pattern.compile(String.join("|", keyList));
Map<String, List<String>> map = valueList.stream()
    .collect(Collectors.groupingBy(s -> {
        Matcher m = p.matcher(s);
        if(!m.find()) throw new AssertionError();
        return m.group();
    }));

map.forEach((k,v) -> System.out.println(k+": "+v));

如果键可能包含特殊字符，这些特殊字符可能会被误解为正则表达式构造，则可以将准备代码更改为

Pattern p = Pattern.compile(
    keyList.stream().map(Pattern::quote).collect(Collectors.joining("|")));

collect操作只会为现有值创建组。如果确实需要显示所有键，则可以使用

Map<String, List<String>> map = valueList.stream()
    .collect(Collectors.groupingBy(s -> {
            Matcher m = p.matcher(s);
            if(!m.find()) throw new AssertionError();
            return m.group();
        },
        HashMap::new, // ensure mutable map
        Collectors.toList()
    ));
keyList.forEach(key -> map.putIfAbsent(key, Collections.emptyList()));

或

Pattern p = Pattern.compile(
    keyList.stream().map(Pattern::quote)
           .collect(Collectors.joining("|", ".*(", ").*")));
Map<String, List<String>> map = valueList.stream()
    .map(p::matcher)
    .filter(Matcher::matches)
    .collect(Collectors.groupingBy(m -> m.group(1),
        HashMap::new, // ensure mutable map
        Collectors.mapping(Matcher::group, Collectors.toList())
    ));
keyList.forEach(key -> map.putIfAbsent(key, Collections.emptyList()));

Java8按值中包含的键创建Map分组

1 个答案: