从 Map 反转 HashMap映射<布尔值，列表<字符串>>

Question

是否有更优雅/内置的方法来反转 Hashmap 的键和值？

我目前有以下内容。

private Map<Boolean, List<String>> reverseMap(Map<String, Boolean> permissions) {
    List<String> allow = new ArrayList<String>();
    List<String> deny = new ArrayList<String>();
    Map<Boolean, List<String>> returnvalue = new HashMap<Boolean, List<String>>();

    for (Entry<String, Boolean> entry : permissions.entrySet()) {
        if(entry.getValue()) {
            allow.add(entry.getKey());
        } else {
            deny.add(entry.getKey());
        }
    }

    returnvalue.put(true, allow);
    returnvalue.put(false, deny);
    return returnvalue;
}

Answer 1

Guava 的 BiMap 已经提供了一种反转其键值对的方法。也许您可以将相关 Map 的接口更改为 BiMap，或者使用以下代码：

private BiMap<Boolean, String> reverseMap(Map<String, Boolean> permissions) {
   BiMap<String, Boolean> bimap = HashBiMap.create(permissions);
   return bimap.inverse();
}

Answer 2

您可以考虑使用 Guava 的 Multimap 实现。例如：

private Multimap<Boolean, String> reverseMap(Map<String, Boolean> permissions) {
   Multimap<Boolean, String> multimap = ArrayListMultimap.create();
   for (Map.Entry<String, Boolean> entry : permissions.entrySet()) {
      multimap.put(entry.getValue(), entry.getKey());
   }
   return multimap;
}

或者更一般地说：

private static <K, V> Multimap<V, K> reverseMap(Map<K, V> source) {
   Multimap<V, K> multimap = ArrayListMultimap.create();
   for (Map.Entry<K, V> entry : source.entrySet()) {
      multimap.put(entry.getValue(), entry.getKey());
   }
   return multimap;
}

Answer 3

我会做类似的事情（但如果你必须经常做这种事情，请考虑 Guava），只用 Set 替换 List （似乎更一致）并预填充反向映射：

private Map<Boolean, Set<String>> reverseMap(Map<String, Boolean> permissions) {
    Map<Boolean, Set<String>> returnvalue = new HashMap<Boolean, Set<String>>();
    returnvalue.put(Boolean.TRUE, new HashSet<String>());
    returnvalue.put(Boolean.FALSE, new HashSet<String>());
    for (Entry<String, Boolean> entry : permissions.entrySet()) 
        returnvalue.get(entry.getValue()).add(entry.getKey());
    return returnvalue;
}

Answer 4

首先要注意的是，如果您的值只是 true 或 false，那么您实际上并不需要反向映射。如果你有更广泛的价值观，那就有意义了。

获取具有特定值的条目的一种简单（但不是很优雅）的方法是：

public static <T, E> Set<T> getKeysByValue(Map<T, E> map, E value) {
     Set<T> keys = new HashSet<T>();
     for (Entry<T, E> entry : map.entrySet()) {
         if (entry.getValue().equals(value)) {
             keys.add(entry.getKey());
         }
     }
     return keys;
}

您可以看到，如果您需要时不时地调用它，那么这不太好。拥有两个不同的地图（正向和反向）并向两者添加条目是有意义的。您不能使用 Bidi 映射，因为键和值之间不存在 1:1 关系。

更新：以下解决方案不起作用。请参阅评论。

您还可以考虑使用 TreeMap 并根据值对其进行排序。这样，您可以随时调用 map.entrySet() 来获得排序集（先拒绝条目，然后允许）。缺点是只有一套。

ValueComparator bvc =  new ValueComparator(map);
TreeMap<String,Boolean> sorted_map = new TreeMap(bvc);

class ValueComparator implements Comparator {
  Map base;

  public ValueComparator(Map base) {
      this.base = base;
  }

  public int compare(Object a, Object b) {
    return (Boolean)base.get(a).compareTo((Boolean)base.get(b));
  }
}