Java反转映射

时间:2011-08-22 11:46:16

标签: java map invert

我需要创建逆映射 - 选择唯一值并为它们找到键。 似乎只有迭代所有键/​​值对的方法,因为entrySet返回的set值不是唯一的? 感谢。

9 个答案:

答案 0 :(得分:35)

地图中的值可能不是唯一的。但如果它们(在你的情况下)你可以像你在你的问题中写的那样做,并创建一个通用的方法来转换它:

private static <V, K> Map<V, K> invert(Map<K, V> map) {

    Map<V, K> inv = new HashMap<V, K>();

    for (Entry<K, V> entry : map.entrySet())
        inv.put(entry.getValue(), entry.getKey());

    return inv;
}

Java 8:

public static <V, K> Map<V, K> invert(Map<K, V> map) {
    return map.entrySet()
              .stream()
              .collect(Collectors.toMap(Entry::getValue, Entry::getKey));
}

使用示例:

public static void main(String[] args) {

    Map<String, Integer> map = new HashMap<String, Integer>();

    map.put("Hello", 0);
    map.put("World!", 1);

    Map<Integer, String> inv = invert(map);

    System.out.println(inv); // outputs something like "{0=Hello, 1=World!}"
}

旁注:put(.., ..)方法将返回密钥的“旧”值。如果它不为null,您可以抛出new IllegalArgumentException("Map values must be unique")或类似的东西。

答案 1 :(得分:9)

看看Google Guava BiMap

使用示例

Map<Integer, String> map = new HashMap<>();
map.put(1, "one");
map.put(2, "two");

Map<String, Integer> inverted = HashBiMap.create(map).inverse();

答案 2 :(得分:4)

在java 8中获取给定地图的反转形式:

public static <K, V> Map<V, K> inverseMap(Map<K, V> sourceMap) {
    return sourceMap.entrySet().stream().collect(
        Collectors.toMap(Entry::getValue, Entry::getKey,
           (a, b) -> a) //if sourceMap has duplicate values, keep only first
        );
}

使用示例

Map<Integer, String> map = new HashMap<Integer, String>();

map.put(1, "one");
map.put(2, "two");

Map<String, Integer> inverted = inverseMap(map);

答案 3 :(得分:3)

  

似乎只有迭代所有键/​​值对的方法,因为entrySet返回的set值不唯一?

这至少是一种方式。这是一个例子:

Map<Integer, String> map = new HashMap<Integer, String>();

map.put(1, "one");
map.put(2, "two");

Map<String, Integer> inverted = new HashMap<String, Integer>();

for (Integer i : map.keySet())
    inverted.put(map.get(i), i);

如果是非唯一值,此算法会将找到的最后一个值映射到其键。 (由于大多数地图的迭代顺序未定义,因此这应该与任何解决方案一样好。)

如果您确实想要保留每个键的第一个值,可以将其更改为

if (!inverted.containsKey(map.get(i)))
    inverted.put(map.get(i), i);

答案 4 :(得分:2)

我会给出另一个方法来解决这个问题: 在EntrySet中重复值

public static void main(String[] args) {

    HashMap<Integer, String> s = new HashMap<Integer, String>();
    s.put(1, "Value1");
    s.put(2, "Value2");
    s.put(3, "Value2");
    s.put(4, "Value1");

    /*
     * swap goes here
     */
    HashMap<String,List<Integer>> newMap = new HashMap<String, List<Integer>>();
    for (Map.Entry<Integer, String> en : s.entrySet()) {
        System.out.println(en.getKey() + " " + en.getValue());

        if(newMap.containsKey(en.getValue())){
            newMap.get(en.getValue()).add(en.getKey());
        } else {
            List<Integer> tmpList = new ArrayList<Integer>();
            tmpList.add(en.getKey());
            newMap.put(en.getValue(), tmpList);
        }
    }

    for(Map.Entry<String, List<Integer>> entry: newMap.entrySet()){
        System.out.println(entry.getKey() + " " + entry.getValue());
    }
}

结果将是:

  

1 Value1
2 Value2
3 Value2
4 Value1
Value1 [1,4]
Value2 [2,3]

答案 5 :(得分:1)

Apache Commons Collections还为双向地图提供了BidiMap界面,以及多个实现。

BidiMap JavaDoc

答案 6 :(得分:0)

您必须假设值可能相同,因为Map合约允许它。

在我看来,最好的解决方案在于使用包装器。它将包含原始值,并添加一个id。它的hashCode()函数将依赖于id,并为原始值提供Getter。 代码将是这样的:

public class MapKey
{
    /**
     * A new ID to differentiate equal values 
     */
    private int _id;
    /**
     * The original value now used as key
     */
    private String _originalValue;

    public MapKey(String originalValue)
    {
        _originalValue = originalValue;
       //assuming some method for generating ids...
        _id = getNextId();
    }

    public String getOriginalValue()
    {
        return _originalValue;
    }

    @Override
    public int hashCode()
    {
        final int prime = 31;
        int result = 1;
        result = prime * result + _id;
        return result;
    }

    @Override
    public boolean equals(Object obj)
    {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        MapKey other = (MapKey) obj;
        if (_id != other._id)
            return false;
        return true;
    }

    @Override
    public String toString()
    {
        StringBuilder sb = new StringBuilder();
        sb.append("MapKey value is ");
        sb.append(_originalValue);
        sb.append(" with ID number ");
        sb.append(_id);
        return sb.toString();
    }

反转地图将是这样的:

public Map <MapKey, Integer> invertMap(Map <Integer, String> map)
{

     Map <MapKey, Integer> invertedMap = new HashMap <MapKey, Integer>();

   Iterator<Entry<Integer, String>> it = map.entrySet().iterator();

   while(it.hasNext())
   {
       //getting the old values (to be reversed)
       Entry<Integer, String> entry = it.next();
       Integer oldKey = entry.getKey();
       String oldValue = entry.getValue();

       //creating the new MapKey
       MapKey newMapKey = new MapKey(oldValue);
       invertedMap.put(newMapKey, oldKey);
   }

   return invertedMap;
}

打印如下所示的值:

for(MapKey key : invertedMap.keySet())
       {
           System.out.println(key.toString() + " has a new value of " +  invertedMap.get(key));

       }

这些代码都没有经过测试,但我相信它是最好的解决方案,因为它使用OO继承设计而不是“c”样式检查,并允许您显示所有原始键和值。

答案 7 :(得分:0)

用番石榴

Multimaps.transformValues(Multimaps.index(map.entrySet(), Map.Entry::getValue),
        Map.Entry::getKey)

您将获得一个多图(基本上是一个列表图)。

答案 8 :(得分:0)

如果您的值重复并且您需要将键存储在列表中,您可以使用

val invertedMap = originalMap.entrySet().stream()
        .collect(Collectors.groupingBy(
                Map.Entry::getValue, 
                Collectors.mapping(Map.Entry::getKey, Collectors.toList()))
        );