LeetCode118|前K个高频单词

示例 1：

输入: ["i", "love", "leetcode", "i", "love", "coding"], k = 2
输出: ["i", "love"]
解析: "i" 和 "love" 为出现次数最多的两个单词，均为2次。
    注意，按字母顺序 "i" 在 "love" 之前。
 

示例 2：

输入: ["the", "day", "is", "sunny", "the", "the", "the", "sunny", "is", "is"], k = 4
输出: ["the", "is", "sunny", "day"]
解析: "the", "is", "sunny" 和 "day" 是出现次数最多的四个单词，
    出现次数依次为 4, 3, 2 和 1 次。
 

注意：

假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
输入的单词均由小写字母组成。

import java.util.*;
import java.util.stream.Collectors;

public class TopKFrequentTest3 {
    public static void main(String[] args) {
        String[] words = {"i", "love", "leetcode", "i", "love", "coding"};
        int k = 2;
        List<String> stringList = topKFrequent(words, k);
        System.out.println("stringList = " + stringList);
    }

    public static List<String> topKFrequent(String[] words, int k) {
        List<String> list = new ArrayList<>();
        if (words == null || k == 0) {
            return list;
        }
        Arrays.sort(words);
        HashMap<String, Integer> hashMap = new LinkedHashMap<>(words.length);
        for (String word : words
        ) {
            hashMap.put(word, hashMap.getOrDefault(word, 0) + 1);
        }

        List<String> stringList = hashMap
                .entrySet()
                .stream()
                .sorted((o1, o2) -> o2.getValue().compareTo(o1.getValue()))
                .collect(Collectors.toList())
                .stream()
                .map(Map.Entry::getKey)
                .collect(Collectors.toList());
        for (int i = 0; i < k; i++) {
            list.add(stringList.get(i));
        }
        return list;
    }
}