如何巧用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,"we"}; Arrays.asList(wordArray).forEach(word ->{ wordCount.putIfAbsent(word,0); wordCount.compute(word,(w,count)->count+1); }); }
當然,我們可以將putIfAbsent放到compute中:
public void countAfter8WithCompute2(){ Map<String,"we"}; Arrays.asList(wordArray).forEach(word -> wordCount.compute(word,count)->count == null ? 1 : count + 1)); }
一行程式碼就完成了。
JDK8中使用merge
再看看merge方法:
default V merge(K key,V value,BiFunction<? 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,"we"}; Arrays.asList(wordArray).forEach(word->wordCount.merge(word,1,(oldCount,one) -> oldCount + one)); }
後面的函式可以用Integer::sum替代:
public void countAfter8WithMerge(){ Map<String,Integer::sum)); }
本文的例子 https://github.com/ddean2009/learn-java-base-9-to-20/tree/master/java-base