乐趣区

Java必修课通过Value获取Map中的键值Key的四种方法

1 简介

我们都知道 Map 是存放键值对 <Key,Value> 的容器,知道了 Key 值,使用方法 Map.get(key) 能快速获取 Value 值。然而,有的时候我们需要反过来获取,知道 Value 值,求 Key 值。

本文将用实例介绍四种方法,通过传入 Value 值,获取得到 Key 值。

2 四种方法

2.1 循环法

循环法就是通过遍历 Map 里的 Entry,一个个比较,把符合条件的找出来。会有三种情况:

  • (1)找到一个值
  • (2)找到多个值
  • (3)找不到

具体代码如下:

@Test
public void loop() {Map<String, Integer> map = ImmutableMap.of("A", 1, "B", 2, "C", 3, "D", 2);
  // 找到一个值
  assertEquals("A", getKeyByLoop(map, 1));
  // 找到多个值
  assertEquals(ImmutableSet.of("B", "D"), getKeysByLoop(map, 2));
  // 找不到
  assertEquals(null, getKeyByLoop(map, 4));
}

private <K, V> K getKeyByLoop(Map<K, V> map, V value) {for (Map.Entry<K, V> entry : map.entrySet()) {if (Objects.equals(entry.getValue(), value)) {return entry.getKey();
    }
  }
  return null;
}

private <K, V> Set<K> getKeysByLoop(Map<K, V> map, V value) {Set<K> set = Sets.newHashSet();
  for (Map.Entry<K, V> entry : map.entrySet()) {if (Objects.equals(entry.getValue(), value)) {set.add(entry.getKey());
    }
  }
  return set;
}

想特别说的一点是,在对比是否相等的时候,使用了 Objects.equals(a, b) 方法,而不是用 a.equals(b) 方法。这样可以避免空指针异常。

2.2 Stream 方法

Stream总是在多种集合操作上都能提供优雅直观的方法,易写易理解。通过一个过滤器,即可把满足相等条件的值取出来,代码如下:

@Test
public void stream() {Map<String, Integer> map = ImmutableMap.of("A", 1, "B", 2, "C", 3, "D", 2);
  assertEquals(ImmutableSet.of("B", "D"), getKeysByStream(map, 2));
}

private <K, V> Set<K> getKeysByStream(Map<K, V> map, V value) {return map.entrySet()
    .stream()
    .filter(kvEntry -> Objects.equals(kvEntry.getValue(), value))
    .map(Map.Entry::getKey)
    .collect(Collectors.toSet());
}

2.3 Guava 的 BiMap

Google 的 Guava 提供了 BiMap 这样一个双向 Map,调用 inverse() 方法会返回一个反向的关联的 BiMap,然后便可以通过get() 方法获取 key 值了。

代码如下:

@Test
public void guava() {BiMap<String, Integer> biMap = HashBiMap.create();
  biMap.put("A", 1);
  biMap.put("B", 2);
  biMap.put("C", null);
  biMap.put("D", 4);
  assertEquals("D", biMap.inverse().get(4));
}

需要注意的是,BiMap作为一个双向的 Map,它不能存储多对一的关系;而HashMap 是可以的。其实很好理解,因为是双向的,所以即要满足 Key 值的唯一性,也要满足 Value 值的唯一性。如果往里存放同样的 Value,会抛异常:java.lang.IllegalArgumentException: value already present

2.4 Apache Commons Collections 的 BidiMap

类似地,Apache Commons Collections也提供了双向 Map 的类 BidiMap,它也是维持一对一的关系,不能多对一。它提供了getKey(value) 方法返回 Key 值。代码如下:

@Test
public void apacheCommons() {BidiMap<String, Integer> bidiMap = new DualHashBidiMap<>();
  bidiMap.put("A", 1);
  bidiMap.put("B", 2);
  bidiMap.put("C", null);
  bidiMap.put("D", 4);
  assertEquals("D", bidiMap.getKey(4));
}

与 Guava 的 BiMap 不同的是,当存放同样的 Value 时,它不会抛异常,而是覆盖原有的数据。

3 总结

本文介绍了四种通过 Value 值获取 Map 中的 Key 值的方法,分别是循环法、Stream、Guava、Apache Commons Collections,这四种方法类似但不尽相同。

  • (1)循环法和使用 Stram 本质上都是要遍历的,如果一个 Map 经常需要反向取 Key 值,则不建议使用,可以考虑 Guava 和 Apache Commons 提供的双向 Map;
  • (2)双向 Map 其实是一种空间换取时间的思想,虽然能较快的找到满足条件的 Key 值,但它也使用了更多的空间来储存双向 Map;
  • (3)双向 Map 并不支持多对一的关系。

如何选择,就看具体需求来取舍了。


欢迎关注公众号 <南瓜慢说>,将持续为你更新 …

多读书,多分享;多写作,多整理。

退出移动版