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