缩小一个映射,其中键是某些条目中的值

时间:2019-01-11 14:26:56

标签: java dictionary hashmap

我正在尝试从另一个地图创建一个新地图,其中某些值是其他条目中的键。

示例:

HashMap<String,String> testMap = new HashMap<>();
testMap.put("a","b");
testMap.put("b","d");
testMap.put("d","e");
testMap.put("e","f");
testMap.put("k","r");

因此,我需要使用以下格式的新地图:

a->f
b->f
d->f
e->f
k->r

producedMap.put("a","f");
producedMap.put("b","f");
producedMap.put("d","f");
producedMap.put("e","f");
producedMap.put("k","r");

我的代码就是这样,但似乎没有给出真实的结果。

    public HashMap<String,String> getMatched(HashMap<String,String> correpondanceMap){

    Collection<String> correpondanceKeys = correpondanceMap.keySet();
    HashMap<String,String> newCorrepondanceMap= new HashMap<>();
    correpondanceMap.entrySet().forEach(entry->{
        if (correpondanceKeys.contains(entry.getValue())){
            String newValue = entry.getValue();
            String keyOfnewValue = correpondanceMap
                    .entrySet()
                    .stream()
                    .filter(entriii -> newValue.equals(entry.getValue()))
                    .map(Map.Entry::getKey).limit(1).collect(Collectors.joining());


            newCorrepondanceMap.put(keyOfnewValue,correpondanceMap.get(newValue));
        }
        else
        {
            newCorrepondanceMap.put(entry.getKey(),entry.getValue());
        }
    });

    newCorrepondanceMap.entrySet().forEach(entry-> System.out.println(entry.getKey() +"  -- > " +entry.getValue()));

    return newCorrepondanceMap;
}

1 个答案:

答案 0 :(得分:10)

您可以通过辅助函数中的一些简单的递归逻辑来实现:

public static String findCorrespondingValue(Map<String, String> map, String key){
    if(map.containsKey(key)){
        return findCorrespondingValue(map, map.get(key));
    }
    return key;
}

如前所述,逻辑很简单,我们只检查给定key中是否存在给定map的值

  • 如果是,我们然后再次执行该函数,但是这次以value作为 新的key
  • 如果不存在映射,我们可以放心地说,给出的key是最后一个 链中的价值

您可以这样调用方法:

Map<String, String> testMap = ... // setup testMap

Map<String, String> result = new HashMap<>();
for (final Entry<String, String> entry : testMap.entrySet()) {
    result.put(
        entry.getKey(), 
        findCorrespondingValue(testMap, entry.getValue())
    );
}

或者如果您碰巧使用了Java 8:

Map<String, String> result = testMap.entrySet().stream()
    .collect(Collectors.toMap(
        e -> e.getKey(),  // or just Map.Entry::getKey
        e -> findCorrespondingValue(e.getValue())
     ));

您当然必须实现某种逻辑,以查明是否具有循环引用。例如:

a -> b
b -> f
f -> a

当前会失败的原因是StackOverflowError


如果您想支持多种不同类型,而不仅仅是String,则可以使它通用:

public static <T> T findCorrespondingValue(Map<? extends T, ? extends T> map, T key){
    if(map.containsKey(key)){
        return findCorrespondingValue(map, map.get(key));
    }
    return key;
}