談談java中遍歷Map的幾種方法

java中的map遍歷有多種方法，從最早的Iterator，到java5支持的foreach,再到java8 Lambda，讓我們一起來看下具體的用法以及各自的優缺點

先初始化一個map

public class TestMap {
  public static Map<Integer, Integer> map = new HashMap<Integer, Integer>();
}

keySet values

如果只需要map的key或者value，用map的keySet或values方法無疑是最方便的

  // KeySet 獲取key
  public void testKeySet() {
    for (Integer key : map.keySet()) {
      System.out.println(key);
    }
  }
  // values 獲取value
  public void testValues() {
    for (Integer value : map.values()) {
      System.out.println(value);
    }
  }

keySet get(key)

如果需要同時獲取key和value，可以先獲取key,然後再通過map的get(key)獲取value

需要說明的是，該方法不是最優選擇，一般不推薦使用

  // keySet get(key) 獲取key and value
  public void testKeySetAndGetKey() {
    for (Integer key : map.keySet()) {
      System.out.println(key + ":" + map.get(key));
    }
  }

entrySet

通過對map entrySet的遍歷，也可以同時拿到key和value，一般情況下，性能上要優於上一種,這一種也是最常用的遍歷方法

  // entrySet 獲取key and value
  public void testEntry() {
    for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
      System.out.println(entry.getKey() + ":" + entry.getValue());
    }
  }

Iterator

對於上面的幾種foreach都可以用Iterator代替，其實foreach在java5中才被支持，foreach的寫法看起來更簡潔

但Iterator也有其優勢：在用foreach遍歷map時，如果改變其大小，會報錯，但如果只是刪除元素，可以使用Iterator的remove方法刪除元素

  // Iterator entrySet 獲取key and value
  public void testIterator() {
    Iterator<Map.Entry<Integer, Integer>> it = map.entrySet().iterator();
    while (it.hasNext()) {
      Map.Entry<Integer, Integer> entry = it.next();
      System.out.println(entry.getKey() + ":" + entry.getValue());
      // it.remove(); 刪除元素
    }
  }

Lambda

java8提供了Lambda表達式支持，語法看起來更簡潔，可以同時拿到key和value，不過，經測試，性能低於entrySet,所以更推薦用entrySet的方式

  // Lambda 獲取key and value
  public void testLambda() {
    map.forEach((key, value) -> {
      System.out.println(key + ":" + value);
    });
  }

簡單性能測試

用10萬條數據，做了一個簡單性能測試，數據類型為Integer，map實現選取HashMap

  static {
    for (int i = 0; i < 100000; i++) {
      map.put(i, 1);
    }
  }

測試結果如下：

KeySet：           392
Values：           320
keySet get(key)：  552
entrySet：         465
entrySet Iterator：508
Lambda：           536

需要說明的是，map存儲的數據類型，map的大小，以及map的不同實現方式都會影響遍歷的性能，所以該測試結果僅供參考

總結

如果只是獲取key，或者value，推薦使用keySet或者values方式

如果同時需要key和value推薦使用entrySet

如果需要在遍歷過程中刪除元素推薦使用Iterator

如果需要在遍歷過程中增加元素，可以新建一個臨時map存放新增的元素，等遍歷完畢，再把臨時map放到原來的map中