删除字符使频率相同

/*
 * 思路：
 * 1. 使用流处理计算字母频率。
 * 2. 统计频率分布。
 * 3. 判断是否满足题目要求的频率情况。
 */

import java.util.Map;
import java.util.stream.Collectors;

public class SolutionStream {
    public boolean equalFrequency(String word) {
        Map<Character, Long> freqMap = word.chars()
                .mapToObj(c -> (char) c)
                .collect(Collectors.groupingBy(c -> c, Collectors.counting()));
        Map<Long, Long> countMap = freqMap.values().stream()
                .collect(Collectors.groupingBy(c -> c, Collectors.counting()));
        if (countMap.size() == 1 && countMap.containsKey(1L)) {
            return true;
        }
        if (countMap.size() == 2) {
            long minFreq = countMap.keySet().stream().min(Long::compare).orElse(0L);
            long maxFreq = countMap.keySet().stream().max(Long::compare).orElse(0L);
            if (minFreq == 1 && countMap.get(minFreq) == 1) {
                return true;
            }
            if (maxFreq == minFreq + 1 && countMap.get(maxFreq) == 1) {
                return true;
            }
        }
        return false;
    }
}