java - 如何使用 Map.merge 将计数器作为值递增？

Question

我正在尝试通过实现 Map.merge() 来重构一种以更有效的方式对 HashMap 中的键进行计数的方法。我怎样才能通过以下代码实现它:

 public Map<Character, Integer> countCharacterUsage(String input) {
    Map<Character, Integer> characterToCountUsage = new WeakHashMap<>();
    for (char symbol : input.toCharArray()) {
        if (characterToCountUsage.containsKey(symbol)) {
            int symbolsCount = characterToCountUsage.get(symbol);
            characterToCountUsage.put(symbol, ++symbolsCount);
        } else {
            characterToCountUsage.put(symbol, 1);
        }
    }

    return characterToCountUsage;
}

Answer 1

您可以使用compute

characterToCountUsage.compute(symbol, (k,v)-> v==null ? 1 : v+1);

还有merge

characterToCountUsage.merge(symbol,1, (k,v)-> v+1);