Инвертная карта Java

Мне нужно создать инверсную карту - выбрать уникальные значения и для них найти ключи. Кажется, что единственный способ-перебрать все пары ключ / значение, потому что entrySet возвращает набор значений so, не уникальный? Спасибо.

8 ответов


значения на карте могут быть не уникальными. Но если они (в вашем случае), вы можете сделать так, как вы написали в своем вопросе, и создать общий метод для его преобразования:

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, c -> c.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") или что-то подобное.


посмотри 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();

чтобы получить перевернутую форму данной карты в 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);

кажется, что единственный способ-перебрать все пары ключ / значение, потому что entrySet возвращает набор значений so, не уникальный?

Это один из способов, по крайней мере. Вот пример:

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);

коллекции Apache Commons также предоставляют BidiMap интерфейс для двунаправленной карт, наряду с несколькими реализациями.

BidiMap Документации


Я бы дал другой подход к этой проблеме дает дополнительное измерение: повторяющиеся значения в 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]


вы должны предположить, что значения могут быть идентичными, так как контракт карты позволяет это.

на мой взгляд, лучшее решение заключается в использовании оболочку. Он будет содержать исходное значение и добавить идентификатор. Его функция hashCode () будет зависеть от идентификатора, и вы предоставляете Геттер для исходного значения. Код будет примерно таким:

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" и позволяет отображать все исходные ключи и значения.


С Гуавой

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

вы получите multimap (в основном карту списков) взамен.