巧用HashMap一行代碼統計單詞出現次數

簡介

JDK是在一直在疊代更新的，很多我們熟悉的類也悄悄的添加了一些新的方法特性。比如我們最常用的HashMap。

今天給大家講一下HashMap在JDK8中添加的兩個新方法compute和merge，進而實作一行代碼實作單詞統計的功能。一起來看看吧。

愛在JDK8之前

JDK8為我們引入了很多非常非常有用新特性，比如Stream和lambda表達式，可以讓我們的程式更加簡潔。

如果我們需要統計一個數組中單詞出現的次數該怎麼做呢？

這裡不是講算法，是以可以直接使用HashMap：

public void countBefore8(){
        Map<String,Integer> wordCount=  new HashMap<>();
        String[] wordArray= new String[]{"we","are","the","world","we"};
        for(String word: wordArray){
            //如果存在則加1，否則将值設定為1
            if(wordCount.containsKey(word)) {
                wordCount.put(word, wordCount.get(word) + 1);
            }else{
                wordCount.put(word, 1);
            }
        }
    }           

基本上流程是上面樣子的。我們對數組進行周遊，然後判斷這個單詞是否存在于hashMap中，如果存在則+1。

邏輯很簡單，但是看起來有些臃腫。

别怕，我們有JDK8。

JDK8中使用compute

先看下JDK8中compute的定義：

default V compute(K key,
            BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        V oldValue = get(key);

        V newValue = remappingFunction.apply(key, oldValue);
        if (newValue == null) {
            // delete mapping
            if (oldValue != null || containsKey(key)) {
                // something to remove
                remove(key);
                return null;
            } else {
                // nothing to do. Leave things as they were.
                return null;
            }
        } else {
            // add or replace old mapping
            put(key, newValue);
            return newValue;
        }
    }           

可以看到compute有第二個參數BiFunction，BiFunction就是一個函數，輸入兩個參數，傳回一個參數。

BiFunction的兩個參數分别是key和key所對應的oldValue。

可考慮到我們的單詞統計，我們可以直接将oldValue+1 即可。是以使用compute，可以将方法改寫為：

public void countAfter8WithCompute(){
        Map<String,Integer> wordCount=  new HashMap<>();
        String[] wordArray= new String[]{"we","are","the","world","we"};
        Arrays.asList(wordArray).forEach(word ->{
            wordCount.putIfAbsent(word,0);
            wordCount.compute(word,(w,count)->count+1);
        });
    }           

當然，我們可以将putIfAbsent放到compute中：

public void countAfter8WithCompute2(){
        Map<String,Integer> wordCount=  new HashMap<>();
        String[] wordArray= new String[]{"we","are","the","world","we"};
        Arrays.asList(wordArray).forEach(word -> wordCount.compute(word,(w, count)->count == null ? 1 : count + 1));
    }           

一行代碼就完成了。

JDK8中使用merge

再看看merge方法：

default V merge(K key, V value,
            BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        Objects.requireNonNull(value);
        V oldValue = get(key);
        V newValue = (oldValue == null) ? value :
                   remappingFunction.apply(oldValue, value);
        if (newValue == null) {
            remove(key);
        } else {
            put(key, newValue);
        }
        return newValue;
    }           

merge方法需要3個參數，第一個參數是key，第二個參數是key對應的oldValue為空的值，也就是為空的預設值，第三個參數是一個BiFunction參數。

不同的是BiFunction的第一個參數是oldValue，第二個參數是value。

生成newValue的邏輯是：如果oldValue不存在，則使用value。如果oldValue存在，則調用BiFunction對oldValue和Value進行合并。

我們可以寫出相應的代碼如下：

public void countAfter8WithMerge(){
        Map<String,Integer> wordCount=  new HashMap<>();
        String[] wordArray= new String[]{"we","are","the","world","we"};
        Arrays.asList(wordArray).forEach(word->wordCount.merge(word, 1, (oldCount, one) -> oldCount + one));
    }           

後面的函數可以用Integer::sum替代：

public void countAfter8WithMerge(){
        Map<String,Integer> wordCount=  new HashMap<>();
        String[] wordArray= new String[]{"we","are","the","world","we"};
        Arrays.asList(wordArray).forEach(word->wordCount.merge(word, 1, Integer::sum));
    }           

本文的例子

https://github.com/ddean2009/learn-java-base-9-to-20/tree/master/java-base

本文已收錄于 http://www.flydean.com/wordcount-in-one-line/

最通俗的解讀，最深刻的幹貨，最簡潔的教程，衆多你不知道的小技巧等你來發現！

歡迎關注我的公衆号:「程式那些事」,懂技術，更懂你！