前K个高频单词

/*
思路：
1. 使用流API处理数组。
2. 首先统计每个单词的频率。
3. 然后按频率降序排列，如果频率相同则按字典序升序排列。
4. 最后返回前k个元素。
*/
import java.util.*;
import java.util.stream.*;
 
public class TopKFrequentWordsStream {
    public List<String> topKFrequent(String[] words, int k) {
        Map<String, Long> count = Arrays.stream(words)
                .collect(Collectors.groupingBy(w -> w, Collectors.counting()));
 
        return count.entrySet().stream()
                .sorted((a, b) -> {
                    if (a.getValue().equals(b.getValue())) {
                        return a.getKey().compareTo(b.getKey());
                    }
                    return b.getValue().compareTo(a.getValue());
                })
                .limit(k)
                .map(Map.Entry::getKey)
                .collect(Collectors.toList());
    }
}