【Java必修課】通過Value獲取Map中的鍵值Key的四種方法

  • 2019 年 11 月 3 日
  • 筆記

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並不支援多對一的關係。

如何選擇,就看具體需求來取捨了。


歡迎關注公眾號<南瓜慢說>,將持續為你更新…

多讀書,多分享;多寫作,多整理。