快捷導(dǎo)航

如何巧用HashMap一行代碼統(tǒng)計(jì)單詞出現(xiàn)次數(shù)詳解

更新時(shí)間：2020年07月19日 15:28:19 作者：flydean

這篇文章主要給大家介紹了關(guān)于如何巧用HashMap一行代碼統(tǒng)計(jì)單詞出現(xiàn)次數(shù)的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧

前言

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

今天給大家講一下HashMap在JDK8中添加的兩個(gè)新方法compute和merge，從而實(shí)現(xiàn)一行代碼實(shí)現(xiàn)單詞統(tǒng)計(jì)的功能。一起來看看吧。

愛在JDK8之前

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

如果我們需要統(tǒng)計(jì)一個(gè)數(shù)組中單詞出現(xiàn)的次數(shù)該怎么做呢？

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

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

基本上流程是上面樣子的。我們對(duì)數(shù)組進(jìn)行遍歷，然后判斷這個(gè)單詞是否存在于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有第二個(gè)參數(shù)BiFunction，BiFunction就是一個(gè)函數(shù)，輸入兩個(gè)參數(shù)，返回一個(gè)參數(shù)。

BiFunction的兩個(gè)參數(shù)分別是key和key所對(duì)應(yīng)的oldValue。

可考慮到我們的單詞統(tǒng)計(jì)，我們可以直接將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);
  });
 }

當(dāng)然，我們可以將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個(gè)參數(shù)，第一個(gè)參數(shù)是key，第二個(gè)參數(shù)是key對(duì)應(yīng)的oldValue為空的值，也就是為空的默認(rèn)值，第三個(gè)參數(shù)是一個(gè)BiFunction參數(shù)。

不同的是BiFunction的第一個(gè)參數(shù)是oldValue，第二個(gè)參數(shù)是value。

生成newValue的邏輯是：如果oldValue不存在，則使用value。如果oldValue存在，則調(diào)用BiFunction對(duì)oldValue和Value進(jìn)行合并。

我們可以寫出相應(yīng)的代碼如下：

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));
 }

后面的函數(shù)可以用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

您可能感興趣的文章: