HashMap有幾種遍歷方法?推薦使用哪種?

本文已收錄《面試精選》系列,Gitee 開源地址://gitee.com/mydb/interview

HashMap 的遍歷方法有很多種,不同的 JDK 版本有不同的寫法,其中 JDK 8 就提供了 3 種 HashMap 的遍歷方法,並且一舉打破了之前遍歷方法「很臃腫」的尷尬。

1.JDK 8 之前的遍歷

JDK 8 之前主要使用 EntrySet 和 KeySet 進行遍歷,具體實現程式碼如下。

1.1 EntrySet 遍歷

EntrySet 是早期 HashMap 遍歷的主要方法,其實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    for (Map.Entry<String, String> entry : map.entrySet()) {
        System.out.println(entry.getKey() + ":" + entry.getValue());
    }
}

以上程式的執行結果,如下圖所示:
image.png

1.2 KeySet 遍歷

KeySet 的遍歷方式是循環 Key 內容,再通過 map.get(key) 獲取 Value 的值,具體實現如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    for (String key : map.keySet()) {
        System.out.println(key + ":" + map.get(key));
    }
}

以上程式的執行結果,如下圖所示:
image.png

KeySet 性能問題

通過以上程式碼,我們可以看出使用 KeySet 遍歷,其性能是不如 EntrySet 的,因為 KeySet 其實循環了兩遍集合,第一遍循環是循環 Key,而獲取 Value 有需要使用 map.get(key),相當於有循環了一遍集合,所以 KeySet 循環不能建議使用,因為循環了兩次,效率比較低

1.3 EntrySet 迭代器遍歷

EntrySet 和 KeySet 除了以上直接循環外,我們還可以使用它們的迭代器進行循環,如 EntrySet 的迭代器實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    Iterator<Map.Entry<String, String>> iterator = map.entrySet().iterator();
    while (iterator.hasNext()) {
        Map.Entry<String, String> entry = iterator.next();
        System.out.println(entry.getKey() + ":" + entry.getValue());
    }
}

以上程式的執行結果,如下圖所示:
image.png

1.4 KeySet 迭代器遍歷

KeySet 也可以使用迭代器的方式進行遍歷,實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    Iterator<String> iterator = map.keySet().iterator();
    while (iterator.hasNext()) {
        String key = iterator.next();
        System.out.println(key + ":" + map.get(key));
    }
}

以上程式的執行結果,如下圖所示:
image.png
雖然 KeySet 循環方式不推薦使用,但還是有必要了解一下的。

1.5 迭代器的作用

既然能直接遍歷,那為什麼還要用迭代器呢?通過以下例子我們就知道了。

不使用迭代器刪除

如果不使用迭代器,假如我們在遍歷 EntrySet 時,在遍歷程式碼中刪除元素,程式碼的實現如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    for (Map.Entry<String, String> entry : map.entrySet()) {
        if ("Java".equals(entry.getKey())) {
            // 刪除此項
            map.remove(entry.getKey());
            continue;
        }
        System.out.println(entry.getKey() + ":" + entry.getValue());
    }
}

以上程式的執行結果,如下圖所示:
image.png
可以看到,如果在遍歷的程式碼中動態刪除元素,非迭代器的方式就會報錯。

使用迭代器刪除

接下來,我們使用迭代器循環 EntrySet,並且在循環中動態刪除元素,實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    Iterator<Map.Entry<String, String>> iterator = map.entrySet().iterator();
    while (iterator.hasNext()) {
        Map.Entry<String, String> entry = iterator.next();
        if ("Java".equals(entry.getKey())) {
            // 刪除此項
            iterator.remove();
            continue;
        }
        System.out.println(entry.getKey() + ":" + entry.getValue());
    }
}

以上程式的執行結果,如下圖所示:
image.png
從上述結果可以看出,使用迭代器的優點是可以在循環的時候,動態的刪除集合中的元素。而上面非迭代器的方式則不能在循環的過程中刪除元素(程式會報錯)。

2.JDK 8 之後的遍歷

在 JDK 8 之後 HashMap 的遍歷就變得方便很多了,JDK 8 中包含了以下 3 種遍歷方法:

  • 使用 Lambda 遍歷
  • 使用 Stream 單執行緒遍歷
  • 使用 Stream 多執行緒遍歷

我們分別來看。

2.1 Lambda 遍歷

使用 Lambda 表達式的遍歷方法實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    
    // 循環遍歷
    map.forEach((key, value) -> {
        System.out.println(key + ":" + value);
    });
}

以上程式的執行結果,如下圖所示:
image.png

2.2 Stream 單執行緒遍歷

Stream 遍歷是先得到 map 集合的 EntrySet,然後再執行 forEach 循環,實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    
    // 循環遍歷
    map.entrySet().stream().forEach((entry) -> {
        System.out.println(entry.getKey() + ":" + entry.getValue());
    });
}

以上程式的執行結果,如下圖所示:
image.png

2.3 Stream 多執行緒遍歷

Stream 多執行緒的遍歷方式和上一種遍歷方式類似,只是多執行了一個 parallel 並發執行的方法,此方法會根據當前的硬體配置生成對應的執行緒數,然後再進行遍歷操作,實現程式碼如下:

public static void main(String[] args) {
    // 創建並賦值 hashmap
    HashMap<String, String> map = new HashMap() {{
        put("Java", " Java Value.");
        put("MySQL", " MySQL Value.");
        put("Redis", " Redis Value.");
    }};
    // 循環遍歷
    map.entrySet().stream().parallel().forEach((entry) -> {
        System.out.println(entry.getKey() + ":" + entry.getValue());
    });
}

以上程式的執行結果,如下圖所示:
image.png
注意上述圖片的執行結果,可以看出當前執行結果和之前的所有遍歷結果都不一樣(列印元素的順序不一樣),因為程式是並發執行的,所以沒有辦法保證元素的執行順序和列印順序,這就是並發編程的特點。

推薦使用哪種遍歷方式?

不同的場景推薦使用的遍歷方式是不同的,例如,如果是 JDK 8 之後的開發環境,推薦使用 Stream 的遍歷方式,因為它足夠簡潔;而如果在遍歷的過程中需要動態的刪除元素,那麼推薦使用迭代器的遍歷方式;如果在遍歷的時候,比較在意程式的執行效率,那麼推薦使用 Stream 多執行緒遍歷的方式,因為它足夠快。所以這個問題的答案是不固定的,我們需要知道每種遍歷方法的優缺點,再根據不同的場景靈活變通。

總結

本文介紹了 7 種 HashMap 的遍歷方式,其中 JDK 8 之前主要使用 EntrySet 和 KeySet 的遍歷方式,而 KeySet 的遍歷方式性能比較低,一般不推薦使用。然而在 JDK 8 之後遍歷方式就有了新的選擇,可以使用比較簡潔的 Lambda 遍歷,也可以使用性能比較高的 Stream 多執行緒遍歷。

是非審之於己,毀譽聽之於人,得失安之於數。

部落客介紹:80 後程式設計師,寫部落格這件事「堅持」了 11 年,愛好:讀書、慢跑、羽毛球。

公眾號:Java面試真題解析

Tags: